update

examples/conv_tasnet/step_2_train_model.py

@@ -198,7 +198,7 @@ def main():
     if config.lr_scheduler == "CosineAnnealingLR":
         lr_scheduler = torch.optim.lr_scheduler.CosineAnnealingLR(
             optimizer,
-            last_epoch=-1,
+            last_epoch=last_epoch,
             # T_max=10 * config.eval_steps,
             # eta_min=0.01 * config.lr,
             **config.lr_scheduler_kwargs,
@@ -250,7 +250,7 @@ def main():
         total_mr_stft_loss = 0.
         total_batches = 0.
 
-        total_steps = 0
+        total_steps = 0 if last_steps == -1 else last_steps
         progress_bar_train = tqdm(
             desc="Training; epoch-{}".format(idx_epoch),
         )