Trainer._load_rng_state() path fix (#14069) (#14071)

3fefa292 · Robert Stone · GitHub · 3892d09f · 3fefa292
Unverified Commit 3fefa292 authored Oct 19, 2021 by Robert Stone Committed by GitHub Oct 19, 2021
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
@@ -1511,7 +1511,7 @@ class Trainer:
                return
        else:
            rng_file = os.path.join(checkpoint, "rng_state.pth")
-            if not os.path.isfile(os.path.join(checkpoint, rng_file)):
+            if not os.path.isfile(rng_file):
                logger.info(
                    "Didn't find an RNG file, if you are resuming a training that was launched in a distributed "
                    "fashion, reproducibility is not guaranteed."