How to count the number of zero weights in a pytorch model

vision
4

You could also go for :

zeros += param.numel() - param.nonzero().size(0)

I am not sure which one is the fastest, but I find this way a bit clearer.

3 Likes