From f774b2e8c43ae7a49adde0e6978abda13ac7a82f Mon Sep 17 00:00:00 2001 From: Haebin Shin Date: Thu, 22 Oct 2020 20:55:31 +0900 Subject: [PATCH] support relative path for best_model_checkpoint (#7973) --- src/transformers/trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py index c20799f638..482ea43ef0 100755 --- 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],