Unverified Commit ca7ff64f authored by Stas Bekman's avatar Stas Bekman Committed by GitHub
Browse files

[deepspeed] fix resume from checkpoint (#11352)

This PR fixes a bug that most likely somehow got exposed (not caused) by https://github.com/huggingface/transformers/pull/11318 - surprisingly the same test worked just fine before that other PR.
parent 74712e22
......@@ -1017,7 +1017,7 @@ class Trainer:
"yield to errors or unwanted behaviors."
)
if self.deepspeed:
if args.deepspeed:
# will be resumed in deepspeed_init
pass
else:
......
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