From ef6dd6b8e60871bb766e8c0744e143e4a680aa87 Mon Sep 17 00:00:00 2001 From: Zhen Wang Date: Wed, 16 Sep 2020 10:54:02 +0800 Subject: [PATCH] fix the test_fleet_lars_meta_optimizer ut. (#27291) --- .../fluid/tests/unittests/test_fleet_lamb_meta_optimizer.py | 2 +- .../fluid/tests/unittests/test_fleet_lars_meta_optimizer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_fleet_lamb_meta_optimizer.py b/python/paddle/fluid/tests/unittests/test_fleet_lamb_meta_optimizer.py index ff305fb9523..ec055178d90 100755 --- 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__": diff --git a/python/paddle/fluid/tests/unittests/test_fleet_lars_meta_optimizer.py b/python/paddle/fluid/tests/unittests/test_fleet_lars_meta_optimizer.py index 34ab423e064..0a70710b459 100755 --- 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__": -- GitLab