diff options
Diffstat (limited to 'libs/utils.py')
| -rw-r--r-- | libs/utils.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/libs/utils.py b/libs/utils.py index c019ba9..2f73705 100644 --- a/libs/utils.py +++ b/libs/utils.py @@ -91,6 +91,7 @@ class Trainer(ABC): raise NotImplementedError(f"Unknown log type: '{type(last_metrics)}'") if not inf_mode: num_iters *= len(train_loader) + config.sched_config.warmup_iters *= len(train_loader) # FIXME: a little bit hacky here scheds = dict(self._configure_scheduler( optims.items(), last_iter, num_iters, config.sched_config, )) |