SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

support relative path for best_model_checkpoint (#7973)

Browse Source
This commit is contained in:
Haebin Shin
2020-10-22 20:55:31 +09:00
committed by GitHub
parent 8348105692
commit f774b2e8c4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer.py
View File

@@ -1219,7 +1219,7 @@ class Trainer:
checkpoints_sorted = [checkpoint[1] for checkpoint in checkpoints_sorted] checkpoints_sorted = [checkpoint[1] for checkpoint in checkpoints_sorted]
# Make sure we don't delete the best model. # Make sure we don't delete the best model.
if self.state.best_model_checkpoint is not None: if self.state.best_model_checkpoint is not None:
best_model_index = checkpoints_sorted.index(self.state.best_model_checkpoint) best_model_index = checkpoints_sorted.index(str(Path(self.state.best_model_checkpoint)))
checkpoints_sorted[best_model_index], checkpoints_sorted[-1] = ( checkpoints_sorted[best_model_index], checkpoints_sorted[-1] = (
checkpoints_sorted[-1], checkpoints_sorted[-1],
checkpoints_sorted[best_model_index], checkpoints_sorted[best_model_index],