Merge remote-tracking branch 'origin/master' into save_fp16_model-fal…

…lback
stas00 · Dec 27, 2021 · 2b19c5b · 2b19c5b
2 parents 74f53f0 + c113827
commit 2b19c5b
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
@@ -2004,9 +2004,6 @@ def save_model(self, output_dir: Optional[str] = None):
                     )
                     self.deepspeed.save_checkpoint(output_dir)
 
-            # save a deepspeed checkpoint as well (this is very fast)
-            self.deepspeed.save_checkpoint(output_dir)
-
         elif self.args.should_save:
             self._save(output_dir)