Removed torch.cuda.empty_cache from train loop. (#31530)

8b7cd402 · Nate Cibik · GitHub · 1e79eade · 8b7cd402
Unverified Commit 8b7cd402 authored Jun 21, 2024 by Nate Cibik Committed by GitHub Jun 21, 2024
Show whitespace changes
Inline Side-by-side

Showing with 0 additions and 1 deletion

src/transformers/trainer.py src/transformers/trainer.py +0 -1

No files found.
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -3307,7 +3307,6 @@ class Trainer:
            loss = self.compute_loss(model, inputs)

        del inputs
-        torch.cuda.empty_cache()

        kwargs = {}