fix the test_fleet_lars_meta_optimizer ut. (#27291)

4 years ago · ef6dd6b8e6
parent 4e8582fe5a
commit ef6dd6b8e6
2 changed files with 2 additions and 2 deletions
--- a/python/paddle/fluid/tests/unittests/test_fleet_lamb_meta_optimizer.py
+++ b/python/paddle/fluid/tests/unittests/test_fleet_lamb_meta_optimizer.py
@ -141,7 +141,7 @@ class TestFleetLambMetaOptimizer(unittest.TestCase):
        ops = [op.type for op in avg_cost.block.ops]
        self.assertIn('lamb', ops)
        self.assertIn('cast', ops)
-        self.assertIn('isfinite', ops)
+        self.assertIn('check_finite_and_unscale', ops)


 if __name__ == "__main__":
--- a/python/paddle/fluid/tests/unittests/test_fleet_lars_meta_optimizer.py
+++ b/python/paddle/fluid/tests/unittests/test_fleet_lars_meta_optimizer.py
@ -145,7 +145,7 @@ class TestFleetLarsMetaOptimizer(unittest.TestCase):
        ops = [op.type for op in avg_cost.block.ops]
        self.assertIn('lars_momentum', ops)
        self.assertIn('cast', ops)
-        self.assertIn('isfinite', ops)
+        self.assertIn('check_finite_and_unscale', ops)


 if __name__ == "__main__":