Open github-0-searcher opened 1 year ago
Hi, the reason for this change is 2-fold.
First, there was a recent change in recording module layers for result. After this change, it only records modules that are called (via their respective forward
function).
The second reason is that torchinfo as of now does not record torch functions (such as +
or @
).
In your case, none of the parameters inside ParameterList
is called (with forward
function) and but used only for @
which is not recorded by torchinfo.
Regarding this, there is a new discussion (by me) about the new tracing mechanism that I think would resolve this problem. If you are interested in helping implementation of this, please see here #192 .
Thanks for notification. I will check that out.
Hello. It seems to still be ommited
Hi there. I was trying to inspect mmoe model from mmoe, which has nn.PrameterList.
I was on v1.7.1 and I got something like this.
This seems to be great, nearly all things are included. But nn.ParameterList (w_gates) is omitted. I went through #54 and #84. Seems to be mentioned before and I downgraded it to v1.7.0
I got result as follow, which includes nn.ParameterList, but result itself seems to be incorrect?