计算神经网络的参数量、计算量

from torchstat import stat
import torchvision.models as models

//把模型换成自己需要的
model=models.googlenet()
stat(model, (3, 224, 224))

计算神经网络的参数量、计算量_第1张图片

你可能感兴趣的:(神经网络,深度学习,cnn,pytorch,计算机视觉)