Smarter prediction loop and no- -> no_ in console args (#8151)

* Smarter prediction loop and no- -> no_ in console args * Fix test
2020-10-29 10:56:25 -04:00
parent b0f1c0ee30
commit acf56408d8
5 changed files with 22 additions and 14 deletions
--- a/examples/test_examples.py
+++ b/examples/test_examples.py
@@ -147,7 +147,6 @@ class ExamplesTests(TestCasePlus):
            --num_train_epochs 2
            --output_dir {tmp_dir}
            --overwrite_output_dir
-            --prediction_loss_only
            """.split()

        if torch.cuda.device_count() > 1: