How to calculate the total number of model parameters and floating point calculation？
ToTensor opened this issue · comments
How to calculate the total number of model parameters and floating point calculation？
sum(p.numel() for p in model.parameters() if p.requires_grad)
Something like this should work near this line? I'm not sure about a good method for calculating flops(