From 3c96a993ea1db70a990fc331cad1e3c7a048b04b Mon Sep 17 00:00:00 2001 From: Vishnu Raj Date: Sun, 11 Aug 2024 05:23:39 +0530 Subject: [PATCH] Fix step size in StepLR Constant value is replaced with the argument passed --- examples/mnist/with_argbind_and_refactor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mnist/with_argbind_and_refactor.py b/examples/mnist/with_argbind_and_refactor.py index ec009bd..c434dea 100644 --- a/examples/mnist/with_argbind_and_refactor.py +++ b/examples/mnist/with_argbind_and_refactor.py @@ -149,7 +149,7 @@ def scheduler( gamma : float, optional Learning rate step gamma, by default 0.7 """ - return StepLR(optimizer, step_size=1, gamma=gamma) + return StepLR(optimizer, step_size=step_size, gamma=gamma) @argbind.bind() def main(