Spaces:
Running
Running
update
Browse files
examples/mpnet_aishell/step_2_train_model.py
CHANGED
@@ -171,6 +171,7 @@ def main():
|
|
171 |
for epoch_i in serialization_dir.glob("epoch-*"):
|
172 |
epoch_i = Path(epoch_i)
|
173 |
epoch_idx = epoch_i.stem.split("-")[1]
|
|
|
174 |
if epoch_idx > epoch_max:
|
175 |
epoch_max = epoch_idx
|
176 |
|
|
|
171 |
for epoch_i in serialization_dir.glob("epoch-*"):
|
172 |
epoch_i = Path(epoch_i)
|
173 |
epoch_idx = epoch_i.stem.split("-")[1]
|
174 |
+
epoch_idx = int(epoch_idx)
|
175 |
if epoch_idx > epoch_max:
|
176 |
epoch_max = epoch_idx
|
177 |
|