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
Copy file name to clipboardExpand all lines: README.md
+26-9Lines changed: 26 additions & 9 deletions
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -5,15 +5,32 @@ Usually it is very straightforward to use them on other datasets, but sometimes
5
5
Unfortunately, none of the pytorch repositories with ResNets on CIFAR10 provides an implementation as described in [original paper](https://arxiv.org/abs/1512.03385). If you just use torchvision's models on CIFAR10 you'll get the model **that differs in number of layers and parameters**. That is unacceptable if you want to directly compare ResNets on CIFAR10.
6
6
The purpose of this repo is to provide a valid pytorch implementation of ResNet-s for CIFAR10. Following models are provided:
7
7
8
-
| Name | # layers | # params|
9
-
|-----------|---------:|-------:|
10
-
|ResNet20 | 20 | 0.27M |
11
-
|ResNet32 | 32 | 0.46M |
12
-
|ResNet44 | 44 | 0.66M |
13
-
|ResNet56 | 56 | 0.85M |
14
-
|ResNet110 | 110 | 1.7M |
15
-
|ResNet1202 | 1202 | 19.4M |
8
+
| Name | # layers | # params| Test err(paper) | Test err(this impl.)|
This implementation follows paper in straightforward manner with some caveats. **Firstly**, original paper uses 45k/5k train/validation split to train data, and selects best performing model based on performance on validation set. This implementation does not do any validation testing, so if you need to compare your results on ResNet head-to-head to orginal paper's keep this in mind. **Secondly**, if you want to train ResNet1202 keep in mind that you need 16GB memory on GPU.
0 commit comments