Skip to content

GitLab

Unverified Commit f774b2e8 authored by Haebin Shin's avatar Haebin Shin Committed by GitHub
Browse files

support relative path for best_model_checkpoint (#7973)

parent 83481056
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/trainer.py
View file @ f774b2e8
...@@ -1219,7 +1219,7 @@ class Trainer: ...@@ -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],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment