Change thresh in test (huggingface#27378)

Change thresh
sywangyi · Nov 9, 2023 · e9adb0c · e9adb0c
1 parent 085ea7e
commit e9adb0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/pytorch/test_accelerate_examples.py b/examples/pytorch/test_accelerate_examples.py
@@ -169,7 +169,7 @@ def test_run_ner_no_trainer(self):
         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")))