[CI] Don't test apex (#6021)

2020-07-24 15:34:16 -04:00
parent a884b7fa38
commit c69ea5efc4
2 changed files with 2 additions and 3 deletions
--- a/examples/seq2seq/test_seq2seq_examples.py
+++ b/examples/seq2seq/test_seq2seq_examples.py
@@ -43,7 +43,7 @@ CHEAP_ARGS = {
    "student_decoder_layers": 1,
    "val_check_interval": 1.0,
    "output_dir": "",
-    "fp16": CUDA_AVAILABLE,
+    "fp16": False,  # TODO(SS): set this to CUDA_AVAILABLE if ci installs apex or start using native amp
    "no_teacher": False,
    "fp16_opt_level": "O1",
    "gpus": 1 if CUDA_AVAILABLE else 0,