claudio-unipv / offset-equivariant

Offset equivariant networks
MIT License
6 stars 1 forks source link

Hyperparameters for augmented training using CIFAR-10 dataset #2

Open AdrianSegura220 opened 1 year ago

AdrianSegura220 commented 1 year ago

Dear Claudio,

First, I would like to say that I appreciate you sharing this repository as an implementation of your paper experiments. Moreover, I cloned the repository, and have a question regarding the hyperparameters for the augmented network trained on the CIFAR-10 dataset:

Thank you for your time and help, I hope you can help me out. This would also be great for the community following this paper! :)

AdrianSegura220 commented 1 year ago

Hey @everyone, if you ever have to find hyperparameters that closely resemble the obtained results for the agumented variant of the Resnet on the CIFAR-10 dataset (results present in fig. 2), you can try using the following ones: 'aug_saturation': 0.6333019759452599 'aug_hue': 0.4542140784634742 'aug_brightness': 0.8260613787076381 'aug_contrast': 0.14959625288049752