Spaces:
Running
Running
update
Browse files
examples/mpnet_aishell/step_2_train_model.py
CHANGED
@@ -168,7 +168,9 @@ def main():
|
|
168 |
|
169 |
# resume training
|
170 |
for epoch_i in serialization_dir.glob("epoch-*"):
|
171 |
-
|
|
|
|
|
172 |
exit(0)
|
173 |
|
174 |
# optimizer
|
|
|
168 |
|
169 |
# resume training
|
170 |
for epoch_i in serialization_dir.glob("epoch-*"):
|
171 |
+
epoch_i = Path(epoch_i)
|
172 |
+
epoch_idx = epoch_i.stem.split("-")[1]
|
173 |
+
print(f"epoch_idx: {epoch_idx}")
|
174 |
exit(0)
|
175 |
|
176 |
# optimizer
|