From af30fd4028553c1d1e4e5d389f309f52e046e67d Mon Sep 17 00:00:00 2001 From: johnanthonyjose Date: Mon, 12 Jul 2021 12:13:53 +0800 Subject: [PATCH] Fix Edge Case when iter is at MAX_ITER --- fvcore/common/param_scheduler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fvcore/common/param_scheduler.py b/fvcore/common/param_scheduler.py index bfa3ace..3bc5e1e 100644 --- a/fvcore/common/param_scheduler.py +++ b/fvcore/common/param_scheduler.py @@ -56,7 +56,7 @@ def __init__(self, value: float) -> None: self._value = value def __call__(self, where: float) -> float: - if where >= 1.0: + if (where - self.WHERE_EPSILON) >= 1.0: raise RuntimeError( f"where in ParamScheduler must be in [0, 1]: got {where}" )