"tests/git@developer.sourcefind.cn:chenpangpang/diffusers.git" did not exist on "3fc10ded000d673768bf03195b0600f69af96a50"
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: ...@@ -1024,7 +1024,7 @@ class Trainer:
else: else:
loss.backward() loss.backward()
return loss return loss.detach()
def is_local_master(self) -> bool: 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