Fix translation no_trainer (#23407)

* Fix translation
This commit is contained in:
Zachary Mueller
2023-05-16 13:10:42 -04:00
committed by GitHub
parent 130e154291
commit 8a58809312

View File

@@ -177,7 +177,7 @@ class ExamplesTestsNoTrainer(TestCasePlus):
self.assertTrue(os.path.exists(os.path.join(tmp_dir, "epoch_0")))
self.assertTrue(os.path.exists(os.path.join(tmp_dir, "ner_no_trainer")))
@unittest.skip(reason="Fix me @zack")
@unittest.skip(reason="Fix me @muellerzr")
@mock.patch.dict(os.environ, {"WANDB_MODE": "offline"})
def test_run_squad_no_trainer(self):
tmp_dir = self.get_auto_remove_tmp_dir()
@@ -270,6 +270,7 @@ class ExamplesTestsNoTrainer(TestCasePlus):
--output_dir {tmp_dir}
--max_train_steps=50
--num_warmup_steps=8
--num_beams=6
--learning_rate=3e-3
--per_device_train_batch_size=2
--per_device_eval_batch_size=1