fix lr assert, test=develop (#21780)

5 years ago · 04909137f5
parent 0eb4d990c4
commit 04909137f5
1 changed files with 2 additions and 3 deletions
--- a/python/paddle/fluid/dygraph/learning_rate_scheduler.py
+++ b/python/paddle/fluid/dygraph/learning_rate_scheduler.py
@ -626,9 +626,8 @@ class LinearLrWarmup(LearningRateDecay):
                format(learning_rate))
        self.learning_rate = learning_rate
        self.warmup_steps = warmup_steps
-        assert (end_lr > start_lr,
-                "end_lr {} MUST GREATER than start_lr {}".format(end_lr,
-                                                                 start_lr))
+        assert end_lr > start_lr, "end_lr {} must be greater than start_lr {}".format(
+            end_lr, start_lr)
        self.lr_ratio_before_warmup = (
            float(end_lr) - float(start_lr)) / float(warmup_steps)