Skip to content

GitLab

Unverified Commit e9adb0c9 authored by Zach Mueller's avatar Zach Mueller Committed by GitHub
Browse files

Change thresh in test (#27378) 

Change thresh
parent 085ea7e5
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/pytorch/test_accelerate_examples.py
View file @ e9adb0c9
......@@ -169,7 +169,7 @@ class ExamplesTestsNoTrainer(TestCasePlus):
run_command(self._launch_args + testargs)
result = get_results(tmp_dir)
self.assertGreaterEqual(result["eval_accuracy"], 0.75)
self.assertLess(result["train_loss"], 0.5)
self.assertLess(result["train_loss"], 0.6)
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")))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment