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(),