Update test_ppo_trainer.py

huggingface · Dec 11, 2023 · bf4da73 · bf4da73
1 parent 8f0fc4c
commit bf4da73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_ppo_trainer.py b/tests/test_ppo_trainer.py
@@ -609,7 +609,7 @@ def test_loss_trainer(self):
             apply_mask(advantages[idx], mask[idx]).unsqueeze(0),
             apply_mask(returns[idx], mask[idx]).unsqueeze(0),
         )
-        self.assertAlmostEqual(pg_loss_unmasked.item(), 2.2868, 4)
+        self.assertAlmostEqual(pg_loss_unmasked.item(), 2.0494, 4)
         self.assertAlmostEqual(v_loss_unmasked.item(), 0.09950, 4)
 
     @parameterized.expand(