update riformer mmpretrain

configs/riformer/metafile.yml

@@ -8,7 +8,7 @@ Collections:
         - LayerScale
     Paper:
       URL: https://arxiv.org/abs/xxxx.xxxxx
-      Title: RIFormer: Keep Your Vision Backbone Effective But Removing Token Mixer
+      Title: "RIFormer: Keep Your Vision Backbone Effective But Removing Token Mixer"
     README: configs/riformer/README.md
     Code:
       Version: v1.0.rc6
@@ -179,5 +179,3 @@ Models:
     Converted From:
       Weights:
       Code:
-
-

mmpretrain/models/backbones/__init__.py

@@ -26,7 +26,6 @@ from .mobileone import MobileOne
 from .mobilevit import MobileViT
 from .mvit import MViT
 from .poolformer import PoolFormer
-from .riformer import RIFormer
 from .regnet import RegNet
 from .replknet import RepLKNet
 from .repmlp import RepMLPNet
@@ -37,6 +36,7 @@ from .resnet import ResNet, ResNetV1c, ResNetV1d
 from .resnet_cifar import ResNet_CIFAR
 from .resnext import ResNeXt
 from .revvit import RevVisionTransformer
+from .riformer import RIFormer
 from .seresnet import SEResNet
 from .seresnext import SEResNeXt
 from .shufflenet_v1 import ShuffleNetV1

model-index.yml

@@ -29,6 +29,7 @@ Import:
   - configs/convmixer/metafile.yml
   - configs/densenet/metafile.yml
   - configs/poolformer/metafile.yml
+  - configs/riformer/metafile.yml
   - configs/inception_v3/metafile.yml
   - configs/mvit/metafile.yml
   - configs/edgenext/metafile.yml

tests/test_models/test_backbones/test_riformer.py

@@ -122,7 +122,6 @@ class TestRIFormer(TestCase):
             torch.allclose(feats[i], feats_deploy[i])
         torch.allclose(pred, pred_deploy)
 
-
     def test_structure(self):
         # test drop_path_rate decay
         cfg = deepcopy(self.cfg)

tools/model_converters/reparameterize_model.py

@@ -3,7 +3,6 @@ import argparse
 from pathlib import Path
 
 import torch
-
 from mmcls.apis import init_model
 from mmcls.models.classifiers import ImageClassifier