Skip to content

Commit 4b9ec1b

Browse files
pacman100novice03
authored andcommitted
fix executable batch size issue (huggingface#24067)
* fix executable batch size issue * fix * undo
1 parent fd49683 commit 4b9ec1b

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

src/transformers/trainer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1650,6 +1650,7 @@ def train(
16501650
def _inner_training_loop(
16511651
self, batch_size=None, args=None, resume_from_checkpoint=None, trial=None, ignore_keys_for_eval=None
16521652
):
1653+
self.accelerator.free_memory()
16531654
self._train_batch_size = batch_size
16541655
logger.debug(f"Currently training with a batch size of: {self._train_batch_size}")
16551656
# Data loader and number of training steps

0 commit comments

Comments
 (0)