Pytorch utils

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