diff --git a/scripts/bert/finetune_classifier.py b/scripts/bert/finetune_classifier.py index 31b855798e..b2bb86d063 100644 --- a/scripts/bert/finetune_classifier.py +++ b/scripts/bert/finetune_classifier.py @@ -746,6 +746,6 @@ def evaluate(loader_dev, metric, segment): except AttributeError as e: warnings.warn(e) nlp.utils.version.check_version('1.7.0', warning_only=True, library=mx) - warnings.warn('INT8 Quantization for BERT need mxnet-mkl >= 1.6.0b20200115 or mxnet >= 1.7') + warnings.warn('INT8 Quantization for BERT need mxnet >= 1.7') else: train(task.metrics) diff --git a/scripts/bert/finetune_squad.py b/scripts/bert/finetune_squad.py index ad7faee6de..ba57129279 100644 --- a/scripts/bert/finetune_squad.py +++ b/scripts/bert/finetune_squad.py @@ -897,7 +897,7 @@ def preprocess_dataset(tokenizer, calib_mode) except AttributeError as e: nlp.utils.version.check_version('1.7.0', warning_only=True, library=mx) - warnings.warn('INT8 Quantization for BERT needs mxnet-mkl >= 1.6.0b20200115 or mxnet >= 1.7') + warnings.warn('INT8 Quantization for BERT needs mxnet >= 1.7') warnings.warn(e) elif not only_predict: train()