Pytorch Get Total Number Of Parameters