Update test_ppo_trainer.py

huggingface · Dec 11, 2023 · bf0f705 · bf0f705
1 parent bf4da73
commit bf0f705
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ppo_trainer.py b/tests/test_ppo_trainer.py
@@ -595,7 +595,7 @@ def test_loss_trainer(self):
             returns[idx].unsqueeze(0),
         )
 
-        self.assertAlmostEqual(pg_loss.item(), 2.2868, 4)
+        self.assertAlmostEqual(pg_loss.item(), 2.0494, 4)
         self.assertAlmostEqual(v_loss.item(), 0.09950, 4)
 
         # check if we get same results with masked parts removed