reset cfg

open-mmlab · Jun 11, 2024 · dac02f5 · dac02f5
1 parent dbdec7a
commit dac02f5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/configs/quantization/qat/base/qat_openvino_resnet18_10e_8xb32_in1k.py b/configs/quantization/qat/base/qat_openvino_resnet18_10e_8xb32_in1k.py
@@ -1,4 +1,4 @@
-_base_ = ['mmpretrain::resnet/resnet18_8xb32_in1k.py']
+_base_ = ['mmcls::resnet/resnet18_8xb32_in1k.py']
 
 resnet = _base_.model
 float_checkpoint = 'https://download.openmmlab.com/mmclassification/v0/resnet/resnet18_8xb32_in1k_20210831-fbbb1da6.pth'  # noqa: E501
@@ -18,7 +18,7 @@
     _scope_='mmrazor',
     type='MMArchitectureQuant',
     data_preprocessor=dict(
-        type='mmpretrain.ClsDataPreprocessor',
+        type='mmcls.ClsDataPreprocessor',
         num_classes=1000,
         # RGB format normalization parameters
         mean=[123.675, 116.28, 103.53],
@@ -33,8 +33,8 @@
         tracer=dict(
             type='mmrazor.CustomTracer',
             skipped_methods=[
-                'mmpretrain.models.heads.ClsHead._get_loss',
-                'mmpretrain.models.heads.ClsHead._get_predictions'
+                'mmcls.models.heads.ClsHead._get_loss',
+                'mmcls.models.heads.ClsHead._get_predictions'
             ])))
 
 optim_wrapper = dict(