diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 19:54:37 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2022-08-10 19:54:37 +0800 |
| commit | e7313d916d783744012ac7bb3011469d72803d25 (patch) | |
| tree | 3b51f70c71cab6c0135909f04ae855936698d256 /libs | |
| parent | 81597cdd0a55140f50b32b69507bfa5309b75f44 (diff) | |
Fix epoch scheduler problem
Diffstat (limited to 'libs')
| -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, )) |