Spaces:

qgyd2021
/

nx_denoise

Running

HoneyTian commited on Mar 31

Commit

45e8101

1 Parent(s): 92bf47a

update

Files changed (1) hide show

examples/conv_tasnet/step_2_train_model.py CHANGED Viewed

@@ -252,6 +252,7 @@ def main():
         total_steps = 0 if last_steps == -1 else last_steps
         progress_bar_train = tqdm(
             desc="Training; epoch-{}".format(idx_epoch),
         )
         for train_batch in train_data_loader:

         total_steps = 0 if last_steps == -1 else last_steps
         progress_bar_train = tqdm(
+            initial=total_steps,
             desc="Training; epoch-{}".format(idx_epoch),
         )
         for train_batch in train_data_loader: