数学建模统计模型,统计模型应用场景

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

param.numel()

返回param中元素的数量

统计模型参数量 num_params = sum(param.numel() for param in net.parameters())print(num_params)