[AOT Autograd] Redirect named_parameters to original mod (pytorch#84157)

anijain2305 · pytorchmergebot · commit 0095571135c8 · 2022-08-29T09:11:56.000Z
Helps in comparing accuracy. Pull Request resolved: pytorch#84157 Approved by: https://github.com/Chillee
diff --git a/functorch/functorch/_src/aot_autograd.py b/functorch/functorch/_src/aot_autograd.py
@@ -892,6 +892,7 @@ def forward(*args):
             )
 
     forward.zero_grad = mod.zero_grad
+    forward.named_parameters = mod.named_parameters
     return forward
 
 

Original file line number	Diff line number	Diff line change
`@@ -892,6 +892,7 @@ def forward(*args):`
`892`	`892`	`)`
`893`	`893`
`894`	`894`	`forward.zero_grad = mod.zero_grad`
	`895`	`+ forward.named_parameters = mod.named_parameters`
`895`	`896`	`return forward`
`896`	`897`
`897`	`898`