Pytorch 统计模型参数量 param.numel()

Pytorch 统计模型参数量

    • param.numel()
    • 统计模型参数量

param.numel()

返回param中元素的数量
Pytorch 统计模型参数量 param.numel()_第1张图片

统计模型参数量

num_params = sum(param.numel() for param in net.parameters())
print(num_params)

你可能感兴趣的:(Pytorch,python)