[Bug] Fix torch2 error (#536)

fix torch2 error
open-mmlab · May 17, 2023 · a578fad · a578fad
1 parent 7586af0
commit a578fad
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/mmrazor/structures/quantization/backend_config/common_operator_config_utils.py b/mmrazor/structures/quantization/backend_config/common_operator_config_utils.py
@@ -49,7 +49,9 @@
     'relu_qat', 'bn_qat', 'bn_relu_qat', 'func'
 ])
 
-if digit_version(torch.__version__) >= digit_version('1.13.0'):
+if digit_version(
+        torch.__version__) >= digit_version('1.13.0') and digit_version(
+            torch.__version__) <= digit_version('1.13.1'):
     _Conv1dMetadata = _ConvMetadata(
         nn.Conv1d, nn.ConvTranspose1d, nn.BatchNorm1d, nnqr.Conv1d,
         nnqr.ConvTranspose1d, nni.ConvReLU1d, nni.ConvBn1d, nni.ConvBnReLU1d,

diff --git a/mmrazor/structures/quantization/backend_config/mapping.py b/mmrazor/structures/quantization/backend_config/mapping.py
@@ -7,7 +7,9 @@
 from .openvino import get_openvino_backend_config
 from .tensorrt import get_tensorrt_backend_config
 
-if digit_version(torch.__version__) >= digit_version('1.13.0'):
+if digit_version(
+        torch.__version__) >= digit_version('1.13.0') and digit_version(
+            torch.__version__) <= digit_version('1.13.1'):
     BackendConfigs = {
         'academic': get_academic_backend_config(),
         'native': get_native_backend_config(),