You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Update CIFAR10 README.md to address the issue in loading old checkpoints
Added note on how to load old CIFAR10 weights using the refactored model.
Basically the 'module.' prefix pertaining to DataParallel wrapper needs to be removed (or the model needs to be wrapped in DataParallel), and more importantly, the extra dropout layers need to be removed except for the ones after maxpooling layers.
These are explained in the
Note 1: the initial learning rate, and optimization policy is hard coded just like caffe.
54
-
Note 2: for testing the cifar10/100 weights located in the repository, use the `simplenet_cifar_5m_extra_pool` model instead. see [issue #5](https://github.com/Coderx7/SimpleNet_Pytorch/issues/5) for more information.
55
-
53
+
Note 1: The initial learning rate, and optimization policy is hard coded just like caffe.
54
+
Note 2: For testing the cifar10/100 weights located in the repository, use the `simplenet_cifar_5m_extra_pool` model instead. see [issue #5](https://github.com/Coderx7/SimpleNet_Pytorch/issues/5) and [Error loading checkpoint#7](https://github.com/Coderx7/SimpleNet_Pytorch/issues/7#issuecomment-1932491193) for more information.
56
55
57
56
## Citation
58
57
If you find SimpleNet useful in your research, please consider citing:
0 commit comments