riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-03 21:00:08 +06:00
Code Issues Actions 25 Packages Projects Releases Wiki Activity

[deepspeed] fix resume from checkpoint (#11352)

Browse Source 
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.
This commit is contained in:
Stas Bekman 2021-04-21 07:48:15 -07:00 committed by GitHub
parent 74712e22f3
commit ca7ff64f5b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/trainer.py
View File

@ -1017,7 +1017,7 @@ class Trainer:
"yield to errors or unwanted behaviors." "yield to errors or unwanted behaviors."
) )
if self.deepspeed: if args.deepspeed:
# will be resumed in deepspeed_init # will be resumed in deepspeed_init
pass pass
else: else: