Remove nightly version of mxnet from warning

dmlc · Apr 29, 2021 · 2a52674 · 2a52674
1 parent ad91858
commit 2a52674
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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()