diff --git a/README.md b/README.md index 460bad5df09eb345d79762a3d60b8c1d4f64fdf8..85d936b1156a5debce641292c56a2758a463dabe 100644 --- a/README.md +++ b/README.md @@ -62,6 +62,7 @@ export PYTHONPATH=/path/to/models:$PYTHONPATH | ResNet101 | 77.944 | 93.844 | | ResNet152 | 78.582 | 94.130 | | ResNeXt50 32x4d | 77.592 | 93.644 | +| ResNeXt101 32x8d| 79.520 | 94.586 | | ShuffleNetV2 x0.5 | 60.696 | 82.190 | | ShuffleNetV2 x1.0 | 69.372 | 88.764 | | ShuffleNetV2 x1.5 | 72.806 | 90.792 | diff --git a/official/vision/classification/resnet/README.md b/official/vision/classification/resnet/README.md index c418db18b44cf5a4d6cd96fb2d15ab56da3077c9..49b036af44c8142268d587844ff1ea22e334823d 100644 --- a/official/vision/classification/resnet/README.md +++ b/official/vision/classification/resnet/README.md @@ -14,6 +14,7 @@ | ResNet101 | 77.944 | 93.844 | | ResNet152 | 78.582 | 94.130 | | ResNeXt50 32x4d | 77.592 | 93.644 | +| ResNeXt101 32x8d| 79.520 | 94.586 | 用户可以通过`megengine.hub`直接加载本目录下定义好的模型,例如: diff --git a/official/vision/classification/resnet/model.py b/official/vision/classification/resnet/model.py index 8321082e6a0c3e10ec2478aa493ad03bb46c6859..7689e0e73555f57026e3b133263d4b0f09ee01d8 100644 --- a/official/vision/classification/resnet/model.py +++ b/official/vision/classification/resnet/model.py @@ -367,6 +367,9 @@ def resnext50_32x4d(**kwargs): return ResNet(Bottleneck, [3, 4, 6, 3], **kwargs) +@hub.pretrained( + "https://data.megengine.org.cn/models/weights/resnext101_32x8d_fbaug_79520_80efb344.pkl" +) def resnext101_32x8d(**kwargs): r"""ResNeXt-101 32x8d model from `"Aggregated Residual Transformation for Deep Neural Networks" `_