-
Notifications
You must be signed in to change notification settings - Fork 1.5k
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
How to train #335
Comments
See examples/imagenet/main.py |
Do we need to always normalize using the Imagenet mean/std values (normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])) when we are using an Imagenet pretrained Efficientnet model from here for training on our different dataset? I have read that the pretrained models available in Pytorch needs this normalization as a preprocessing step (https://discuss.pytorch.org/t/how-to-preprocess-input-for-pre-trained-networks/683). |
Hello, I want to ask how I can train my own data set now. I don't see the training code
The text was updated successfully, but these errors were encountered: