save last/best model test fix

nyu-mll · May 3, 2021 · d0efb1c · d0efb1c
1 parent cd1c67c
commit d0efb1c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/proj/simple/test_runscript.py b/tests/proj/simple/test_runscript.py
@@ -105,7 +105,7 @@ def test_simple_runscript_save(tmpdir, task_name, model_type):
  data_dir=data_dir,
  hf_pretrained_model_name_or_path=model_type,
  tasks=task_name,
- train_examples_cap=32,
+ train_examples_cap=64,
  train_batch_size=32,
  do_save=True,
  eval_every_steps=1,