forgot to pass warmup to schedule free

2024-05-09 22:18:49 -05:00 · 2024-05-09 22:18:49 -05:00 · c22a177cf8
commit c22a177cf8
parent b6131565ad
1 changed files with 2 additions and 1 deletions
--- a/vall_e/engines/init.py
+++ b/vall_e/engines/init.py
@ -97,7 +97,8 @@ def load_engines(training=True):

 				optimizer = scheduler_class(
 					[ param for name, param in model.named_parameters() if name not in model._cfg.frozen_params ],
-					lr = params['lr']
+					lr = params['lr'],
+					warmup_steps = cfg.hyperparameters.warmup_steps
 				)