本文共 201 字，大约阅读时间需要 1 分钟。

在实验中，经常需要计算网络的参数量和预算量。

参考：

运算量和参数量的计算代码

# Calculate params & FLOPs    from torchsummaryX import summary    dummy_input = torch.zeros(1, 3, 128, 128).cuda()    summary(model.netG, dummy_input)    exit()

转载地址：http://yvajz.baihongyu.com/