boom boom

2020-05-19 15:57:04 -04:00
parent 2ee9388492
commit 51221fbbf8
2 changed files with 3 additions and 1 deletions
@@ -21,4 +21,5 @@ python finetune.py \
 --fp16 \
 --output_dir $OUTPUT_DIR \
 --val_check_interval 0.5 \
+--num_epochs 4 \
 $@
@@ -144,7 +144,8 @@ class TestBartExamples(unittest.TestCase):
            model_type="bart",
            train_batch_size=2,
            eval_batch_size=2,
-            fast_dev_run=True,
+            num_epochs=2,
+            fast_dev_run=False,
            n_gpu=0,
            output_dir=output_dir,
            do_predict=True,