@@ -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
|
||||
|
||||
Reference in New Issue
Block a user