But in old version of pytorch, the grad_fn nearly made a full copy of module layer information, I can still trace the relations. In new version I can’t find the information in grad_fn anymore T_T.  Are these information moved into private variables?
Another potential way is by adding hooks into each module layer, will that be working?