-
Notifications
You must be signed in to change notification settings - Fork 62
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
Regarding transforms #64
Comments
Actually I never test MNIST, let me know if any issue exists. The transform for MNIST is a simple trojanzoo/trojanvision/datasets/imageset.py Lines 56 to 61 in e38f90b
trojanzoo/trojanvision/datasets/imageset.py Lines 141 to 156 in e38f90b
|
And you will get ResNet18 (first convolutional layer compressed version) with 96.5% accuracy. |
ya so I was testing these models manually and the transform that worked for me is transform.Compose([transforms.ToTensor(),transforms.Normalize([0.49139968, 0.48215827, 0.44653124],[0.24703233, 0.24348505, 0.26158768])) To give the expected accuracy. |
I remembered that Without data augment such as random crop and cutout, the model accuracy won’t exceed 92%. But I could be wrong. And if you use my model class and put normalization into the transform, you’d better set the model transform layer mean/std to be 0/1. |
Close this issue if you have no further concern. |
Can somebody please let me know the transforms used in cfir10 and mnist datasets?
Thank you.
The text was updated successfully, but these errors were encountered: