分类：param.numel()

2021
07-01

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

param.numel()返回param中元素的数量统计模型参数量num_params=sum(param.numel()forparaminnet.parameters())print(num_params)补充：Pytorch查看模型参数Pytorch查看模型参数查看利用Pytorch搭建模型的参数,直接看程序importtorch#引入torch.nn并指定别名importtorch.nnasnnimporttorch.nn.functionalasFclassNet(nn.Module):def__init__(self):#nn.Module子类的函数必须在构造函数中执行父类的... 继续阅读 >