HoneyTian commited on
Commit
8c9b2a3
·
1 Parent(s): 4425d40
examples/conv_tasnet/step_2_train_model.py CHANGED
@@ -172,6 +172,7 @@ def main():
172
 
173
  # resume training
174
  last_steps = -1
 
175
  for step_i in serialization_dir.glob("steps-*"):
176
  step_i = Path(step_i)
177
  step_idx = step_i.stem.split("-")[1]
 
172
 
173
  # resume training
174
  last_steps = -1
175
+ last_epoch = -1
176
  for step_i in serialization_dir.glob("steps-*"):
177
  step_i = Path(step_i)
178
  step_idx = step_i.stem.split("-")[1]