Unverified Commit 25afb4ea authored by Stuart Mesham's avatar Stuart Mesham Committed by GitHub
Browse files

fixed trainer tr_loss memory leak (#6999)

* fixed trainer tr_loss memory leak

* detached returned training loss from computation graph in the Trainer class' training_step() method

* Revert "fixed trainer tr_loss memory leak"

This reverts commit 47226e4e
parent 1b76936d
......@@ -1024,7 +1024,7 @@ class Trainer:
else:
loss.backward()
return loss
return loss.detach()
def is_local_master(self) -> bool:
"""
......
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