Skip to content

Commit 4334c57

Browse files
authored
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
1 parent 2370243 commit 4334c57

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

cifar/README.md

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,8 @@ Simply initiate the training like :
5050
`python3 main.py ./data/cifar.python --dataset cifar10 --arch simplenet_cifar_5m --save_path ./snapshots/simplenet --epochs 540 --batch_size 100 --workers 2`
5151

5252

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) 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.
5655

5756
## Citation
5857
If you find SimpleNet useful in your research, please consider citing:

0 commit comments

Comments
 (0)