wassname/baukit
1
0
Fork 0
mirror of https://github.com/wassname/baukit.git synced 2026-06-27 16:14:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

retain_grad on input

Browse Source
This commit is contained in:
Michael J Clark
2024-08-07 21:50:03 +08:00
committed by GitHub
parent 9d51abd51e
commit bbbf111f8d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
baukit/nethook.py
+2 -2
View File
@@ -78,8 +78,8 @@ class Trace(contextlib.AbstractContextManager):
inputs[0] if len(inputs) == 1 else inputs, inputs[0] if len(inputs) == 1 else inputs,
clone=clone, clone=clone,
detach=detach, detach=detach,
retain_grad=False, retain_grad=retain_grad,
) # retain_grad applies to output only. )
if retain_output: if retain_output:
retainer.output = recursive_copy( retainer.output = recursive_copy(
output, clone=clone, detach=detach, retain_grad=retain_grad output, clone=clone, detach=detach, retain_grad=retain_grad