fujiyuu75
4/8/2020 - 10:05 AM

https://stackoverflow.com/questions/49201236/check-the-total-number-of-parameters-in-a-pytorch-model

pytorch_total_params = sum(p.numel() for p in model.parameters() if p.requires_grad)