Skip to content

GitLab

Unverified Commit c69ea5ef authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

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

parent a884b7fa
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
examples/seq2seq/test_bash_script.py
View file @ c69ea5ef
...@@ -55,9 +55,8 @@ def test_train_mbart_cc25_enro_script(): ...@@ -55,9 +55,8 @@ def test_train_mbart_cc25_enro_script():
if CUDA_AVAILABLE: if CUDA_AVAILABLE:
gpus = 1 # torch.cuda.device_count() gpus = 1 # torch.cuda.device_count()
else: else:
bash_script = bash_script.replace("--fp16", "")
gpus = 0 gpus = 0
bash_script = bash_script.replace("--fp16", "")
testargs = ( testargs = (
["finetune.py"] ["finetune.py"]
+ bash_script.split() + bash_script.split()
......
examples/seq2seq/test_seq2seq_examples.py
View file @ c69ea5ef
...@@ -43,7 +43,7 @@ CHEAP_ARGS = { ...@@ -43,7 +43,7 @@ CHEAP_ARGS = {
"student_decoder_layers": 1, "student_decoder_layers": 1,
"val_check_interval": 1.0, "val_check_interval": 1.0,
"output_dir": "", "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, "no_teacher": False,
"fp16_opt_level": "O1", "fp16_opt_level": "O1",
"gpus": 1 if CUDA_AVAILABLE else 0, "gpus": 1 if CUDA_AVAILABLE else 0,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment