mmpretrain/configs/resnet
whcao af83e981ac
[Bug]Fix label smooth bug (#203)
* add convert_to_one_hot

* add test_label_smooth_loss

* add my label_smooth_loss

* fix CELoss bug

* test new label smooth loss

* LabelSmoothLoss downward compatibility

* add some comments

* remove the old version of LabelSmoothLoss

* add some comments

* add some comments

* add some comments

* add label smooth to config
2021-04-13 13:53:56 +08:00
..
README.md Update README.md (#150) 2021-01-25 17:16:57 +08:00
resnet18_b16x8_cifar10.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet18_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet34_b16x8_cifar10.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet34_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet50_b16x8_cifar10.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet50_b16x8_cifar10_mixup.py [Feature] Add "mixup" from Bag of Tricks (#160) 2021-02-25 14:06:58 +08:00
resnet50_b32x8_coslr_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet50_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet50_b32x8_label_smooth_imagenet.py [Bug]Fix label smooth bug (#203) 2021-04-13 13:53:56 +08:00
resnet50_b32x8_mixup_imagenet.py [Feature] Add "mixup" from Bag of Tricks (#160) 2021-02-25 14:06:58 +08:00
resnet50_b64x32_warmup_coslr_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet50_b64x32_warmup_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet50_b64x32_warmup_label_smooth_imagenet.py fix typo in configs (#147) 2021-01-21 14:47:02 +08:00
resnet101_b16x8_cifar10.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet101_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet152_b16x8_cifar10.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnet152_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnetv1d50_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnetv1d101_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00
resnetv1d152_b32x8_imagenet.py Rearrange config files (#116) 2020-12-18 16:59:50 +08:00

README.md

Deep Residual Learning for Image Recognition

Introduction

[ALGORITHM]

@inproceedings{he2016deep,
  title={Deep residual learning for image recognition},
  author={He, Kaiming and Zhang, Xiangyu and Ren, Shaoqing and Sun, Jian},
  booktitle={Proceedings of the IEEE conference on computer vision and pattern recognition},
  pages={770--778},
  year={2016}
}

Results and models

Cifar10

Model Params(M) Flops(G) Top-1 (%) Top-5 (%) Config Download
ResNet-18-b16x8 11.17 0.56 94.72 config model | log
ResNet-34-b16x8 21.28 1.16 95.34 config model | log
ResNet-50-b16x8 23.52 1.31 95.36 config model | log
ResNet-101-b16x8 42.51 2.52 95.66 config model | log
ResNet-152-b16x8 58.16 3.74 95.96 config model | log

ImageNet

Model Params(M) Flops(G) Top-1 (%) Top-5 (%) Config Download
ResNet-18 11.69 1.82 70.07 89.44 config model | log
ResNet-34 21.8 3.68 73.85 91.53 config model | log
ResNet-50 25.56 4.12 76.55 93.15 config model | log
ResNet-101 44.55 7.85 78.18 94.03 config model | log
ResNet-152 60.19 11.58 78.63 94.16 config model | log
ResNetV1D-50 25.58 4.36 77.4 93.66 config model | log
ResNetV1D-101 44.57 8.09 78.85 94.38 config model | log
ResNetV1D-152 60.21 11.82 79.35 94.61 config model | log