support relative path for best_model_checkpoint (#7973)

f774b2e8 · Haebin Shin · GitHub · 83481056 · f774b2e8
Unverified Commit f774b2e8 authored Oct 22, 2020 by Haebin Shin Committed by GitHub Oct 22, 2020
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +1 -1

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -1219,7 +1219,7 @@ class Trainer:
        checkpoints_sorted = [checkpoint[1] for checkpoint in checkpoints_sorted]
        # Make sure we don't delete the best model.
        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[-1],
                checkpoints_sorted[best_model_index],