From 4130973482c5f825c99614b6cabc16baf7486af2 Mon Sep 17 00:00:00 2001 From: Geewook Kim Date: Fri, 29 Jul 2022 15:06:18 +0000 Subject: [PATCH] fix: max_iter with config.max_steps --- lightning_module.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightning_module.py b/lightning_module.py index a8116be5..3cf9ecbc 100755 --- a/lightning_module.py +++ b/lightning_module.py @@ -114,7 +114,7 @@ def configure_optimizers(self): ) if self.config.get("max_steps", None): - max_iter = min(self.config.max_steps, max_iter) + max_iter = min(self.config.max_steps, max_iter) if max_iter is not None else self.config.max_steps assert max_iter is not None optimizer = torch.optim.Adam(self.parameters(), lr=self.config.lr)