HoneyTian commited on
Commit
5bb28a8
·
1 Parent(s): 949a94d
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
- print(epoch_i)
 
 
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