SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix translation no_trainer (#23407)

Browse Source 
* Fix translation
This commit is contained in:
Zachary Mueller
2023-05-16 13:10:42 -04:00
committed by GitHub
parent 130e154291
commit 8a58809312
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/pytorch/test_accelerate_examples.py
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, "epoch_0")))
self.assertTrue(os.path.exists(os.path.join(tmp_dir, "ner_no_trainer"))) 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"}) @mock.patch.dict(os.environ, {"WANDB_MODE": "offline"})
def test_run_squad_no_trainer(self): def test_run_squad_no_trainer(self):
tmp_dir = self.get_auto_remove_tmp_dir() tmp_dir = self.get_auto_remove_tmp_dir()
@@ -270,6 +270,7 @@ class ExamplesTestsNoTrainer(TestCasePlus):
--output_dir {tmp_dir} --output_dir {tmp_dir}
--max_train_steps=50 --max_train_steps=50
--num_warmup_steps=8 --num_warmup_steps=8
--num_beams=6
--learning_rate=3e-3 --learning_rate=3e-3
--per_device_train_batch_size=2 --per_device_train_batch_size=2
--per_device_eval_batch_size=1 --per_device_eval_batch_size=1