How can I get the model trainable parameters information?
Closed this issue · 1 comments
Yuhuoo commented
During training, I noticed that the trainable parameters of the model were printed in the console. Could you please let me know how this is printed? I couldn't find this print code in the project.
YuliangXiu commented
This should be a built-in feature of PyTorch-Lightning. You could check the document of PL for more details.