From a578fad2bc557069bf189fe7e0220da4396c587e Mon Sep 17 00:00:00 2001 From: humu789 <88702197+humu789@users.noreply.github.com> Date: Wed, 17 May 2023 15:59:51 +0800 Subject: [PATCH] [Bug] Fix torch2 error (#536) fix torch2 error --- .../backend_config/common_operator_config_utils.py | 4 +++- mmrazor/structures/quantization/backend_config/mapping.py | 4 +++- 2 files changed, 6 insertions(+), 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 index 0a381d5d0..bea018975 100644 --- 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 index b9cc5372b..0a02ac1b7 100644 --- 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(),