From 9e49961538a9deb3641b03c1c5e088b0b4f8f601 Mon Sep 17 00:00:00 2001 From: Jerry Jiarui XU Date: Sun, 10 Jul 2022 22:05:09 -0700 Subject: [PATCH] Change ConstantParamScheduler to exclude 1 At the end of the training, `where == 1.0`. --- 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 f95350b..e7f8859 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 > 1.0: raise RuntimeError( f"where in ParamScheduler must be in [0, 1]: got {where}" )