diff --git a/models/cv/classification/mvitv2_base/igie/ci/prepare.sh b/models/cv/classification/mvitv2_base/igie/ci/prepare.sh index 9ec3e6418c4ce714f4c0256900dfb601061df9db..cfb0ea177a351c997b8f4931dfb3a403e7eca490 100644 --- a/models/cv/classification/mvitv2_base/igie/ci/prepare.sh +++ b/models/cv/classification/mvitv2_base/igie/ci/prepare.sh @@ -17,7 +17,7 @@ set -x pip3 install -r ../../igie_common/requirements.txt -pip3 install mmcv==1.5.3 mmcls==0.24.0 +pip3 install --no-build-isolation mmcv==1.5.3 mmcls==0.24.0 unzip -q /mnt/deepspark/data/repos/mmpretrain-0.24.0.zip -d ./ python3 ../../igie_common/export_mmcls.py --cfg mmpretrain/configs/mvit/mvitv2-base_8xb256_in1k.py --weight mvitv2-base_3rdparty_in1k_20220722-9c4f0a17.pth --output mvitv2_base.onnx -onnxsim mvitv2_base.onnx mvitv2_base_opt.onnx \ No newline at end of file +onnxsim mvitv2_base.onnx mvitv2_base_opt.onnx