HoneyTian commited on
Commit
f90cb85
·
1 Parent(s): 3188b07
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