Hey all, as part of ablation study, I had removed modules from the forward pass, but kept them in the model initialization. Is it possible to count only parameters that are actually used during the forward pass?
Thanks