Spaces:
Running
Running
update
Browse files
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]
|