Replace assert statement with if condition and ValueError (#13263)

225de5cc · Nishant Prabhu · GitHub · 46554fc1 · 225de5cc
Unverified Commit 225de5cc authored Aug 25, 2021 by Nishant Prabhu Committed by GitHub Aug 25, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/transformers/optimization.py src/transformers/optimization.py +2 -1

No files found.
--- a/src/transformers/optimization.py
+++ b/src/transformers/optimization.py
@@ -201,7 +201,8 @@ def get_polynomial_decay_schedule_with_warmup(
    """
    lr_init = optimizer.defaults["lr"]
-    assert lr_init > lr_end, f"lr_end ({lr_end}) must be be smaller than initial lr ({lr_init})"
+    if not (lr_init > lr_end):
+        raise ValueError(f"lr_end ({lr_end}) must be be smaller than initial lr ({lr_init})")
    def lr_lambda(current_step: int):
        if current_step < num_warmup_steps: