pytorch 计算模型参数量

 sum(torch.numel(parameter) for parameter in model.parameters())

你可能感兴趣的:(Pytorch,pytorch,神经网络,深度学习)