Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Trained without data augmentation? #2

Closed
nandi-zhang opened this issue Mar 22, 2021 · 2 comments
Closed

Trained without data augmentation? #2

nandi-zhang opened this issue Mar 22, 2021 · 2 comments

Comments

@nandi-zhang
Copy link

nandi-zhang commented Mar 22, 2021

Thanks for the great work on the pytorch implementation of NFNet! The accuracies achieved by this implementation are pretty impressive also and I am wondering if these training results were simply derived from the training script, that is, without data augmentation.

@benjs
Copy link
Owner

benjs commented Mar 22, 2021

The results were derived from the pretrained weights, provided by the authors in their original repo :)

As the data augmentation is not implemented yet, the training script should reach the baseline accuracy as provided in the paper (F0: 80%). This, however, is not tested yet. I am currently adding mutli gpu support (see branch distributed) and try to reach the baseline accuracy on AWS. This will unfortunatly take some time as I am writing exams the next weeks :/

@nandi-zhang
Copy link
Author

Got it. Just take your time and good luck on the exams. :))

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants