Enforce determinism for backwards compatibility checker (apache#14463)

vdantu · Mar 31, 2019 · 9db8329 · 9db8329
1 parent 5bfda35
commit 9db8329
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/nightly/model_backwards_compatibility_check/model_backward_compat_checker.sh b/tests/nightly/model_backwards_compatibility_check/model_backward_compat_checker.sh
@@ -27,4 +27,5 @@ cd tests/nightly/model_backwards_compatibility_check
 echo `pwd`
 
 echo '=========================='
-python model_backwards_compat_inference.py
+export MXNET_ENFORCE_DETERMINISM=1
+python model_backwards_compat_inference.py
diff --git a/tests/nightly/model_backwards_compatibility_check/train_mxnet_legacy_models.sh b/tests/nightly/model_backwards_compatibility_check/train_mxnet_legacy_models.sh
@@ -25,6 +25,7 @@ run_models() {
  echo '=========================='
  echo "Running training files and preparing models"
  echo '=========================='
+ export MXNET_ENFORCE_DETERMINISM=1
  python model_backwards_compat_train.py
  echo '=========================='
 }