Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "73efa694e6cc9141ce732ea8215b25b3083298dc"
Unverified Commit 8b7cd402 authored by Nate Cibik's avatar Nate Cibik Committed by GitHub
Browse files

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

parent 1e79eade
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
src/transformers/trainer.py
View file @ 8b7cd402
...@@ -3307,7 +3307,6 @@ class Trainer: ...@@ -3307,7 +3307,6 @@ class Trainer:
loss = self.compute_loss(model, inputs) loss = self.compute_loss(model, inputs)
del inputs del inputs
torch.cuda.empty_cache()
kwargs = {} kwargs = {}
......
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