From e6d278ca2b8d75793a843f655e30f30beba0ca46 Mon Sep 17 00:00:00 2001 From: Ivan Arcuschin Date: Fri, 13 Sep 2024 14:16:11 +0100 Subject: [PATCH] Fix typing --- iit/model_pairs/base_model_pair.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/iit/model_pairs/base_model_pair.py b/iit/model_pairs/base_model_pair.py index c78a196..1d9428a 100644 --- a/iit/model_pairs/base_model_pair.py +++ b/iit/model_pairs/base_model_pair.py @@ -273,7 +273,7 @@ def train( del scheduler_kwargs['final_lr'] if "lr_lambda" not in scheduler_kwargs: - def linear_lr(step): + def linear_lr(step: int) -> float: return 1 - (step / epochs) * (1 - final_lr / initial_lr) scheduler_kwargs["lr_lambda"] = linear_lr