Trace through nn.Module dunder methods for UnspecializedNNModule (#126578)
Summary:
Tracing through `__init__` is important because it initializes (calls STORE_ATTR) on members. By doing that, we kick in the mutation tracking for these objects. So, things like mutating `_modules` etc is tracked automatically.
Fixes https://github.com/pytorch/pytorch/issues/111837
X-link: https://github.com/pytorch/pytorch/pull/126578
Approved by: https://github.com/jansel
Reviewed By: clee2000
Differential Revision: D58451174
Pulled By: anijain2305
fbshipit-source-id: a9dca5754b57247ed6c3bf97dc6ec0f00670c477