From d0efb1c1485abd3dc52d5e94773e08f75bc6d8e8 Mon Sep 17 00:00:00 2001 From: Jesse Swanson Date: Sun, 2 May 2021 21:00:28 -0400 Subject: [PATCH] save last/best model test fix --- tests/proj/simple/test_runscript.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/proj/simple/test_runscript.py b/tests/proj/simple/test_runscript.py index ea97cc53b..ed048e5f6 100644 --- 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,