Pytorch utils

print parameters num

sum(p.numel() for p in model.parameters())

def numel(model):
    return sum(p.numel() for p in model.parameters())