

def count_params(module, trainable_only=True):
    return sum((p.get_value().size for p in module.parameters(trainable_only=p)))

dataset/ETHPy150Open lucasb-eyer/DeepFried2/DeepFried2/utils.py count_params/VarMisuse@34/41 `trainable_only`->`p`