From e69f0baa6158d32595efe7b2d843d189a29d13e7 Mon Sep 17 00:00:00 2001
From: "mingjiang.li" <mingjiang.li@iluvatar.com>
Date: Tue, 21 Jan 2025 18:04:12 +0800
Subject: [PATCH] bugfix: mmpretraino 0.24.0 not compatible with mmcv 2.1.0

Signed-off-by: mingjiang.li <mingjiang.li@iluvatar.com>
---
 models/cv/classification/deit_tiny/ixrt/README.md | 5 +++--
 models/cv/classification/deit_tiny/ixrt/export.py | 2 +-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/models/cv/classification/deit_tiny/ixrt/README.md b/models/cv/classification/deit_tiny/ixrt/README.md
index 7a58584d..1ceea7bf 100644
--- a/models/cv/classification/deit_tiny/ixrt/README.md
+++ b/models/cv/classification/deit_tiny/ixrt/README.md
@@ -34,10 +34,11 @@ Dataset: <https://www.image-net.org/download.php> to download the validation dat
 
 ```bash
 # git clone mmpretrain
-git clone -b v0.24.0 https://github.com/open-mmlab/mmpretrain.git
+git clone --depth 1 -b v1.1.0 https://github.com/open-mmlab/mmpretrain.git
+(cd mmpretrain/ && python3 setup.py develop)
 
 # export onnx model
-python3 export.py --cfg mmpretrain/configs/deit/deit-tiny_pt-4xb256_in1k.py --weight deit-tiny_pt-4xb256_in1k_20220218-13b382a0.pth --output deit_tiny.onnx
+python3 export.py --cfg mmpretrain/configs/deit/deit-tiny_4xb256_in1k.py --weight deit-tiny_pt-4xb256_in1k_20220218-13b382a0.pth --output deit_tiny.onnx
 
 # Use onnxsim optimize onnx model
 onnxsim deit_tiny.onnx deit_tiny_opt.onnx
diff --git a/models/cv/classification/deit_tiny/ixrt/export.py b/models/cv/classification/deit_tiny/ixrt/export.py
index 0078670f..b2f2f142 100644
--- a/models/cv/classification/deit_tiny/ixrt/export.py
+++ b/models/cv/classification/deit_tiny/ixrt/export.py
@@ -15,7 +15,7 @@
 import argparse
 
 import torch
-from mmcls.apis import init_model
+from mmpretrain.apis import init_model
 
 class Model(torch.nn.Module):
     def __init__(self, config_file, checkpoint_file):
-- 
Gitee