美文网首页PyTorch
PyTorch 打印模型

PyTorch 打印模型

作者: SwiftAI | 来源:发表于2018-11-08 15:31 被阅读162次

方式1

print 打印网络的层

import torchsummary
from torchvision.models.resnet import *
net = resnet18().cuda()
print(net)

方式2

torchsummary.summary

net = resnet18()
torchsummary.summary(net, (3, 224, 224))

总结

两种方法打印出来的都是net函数中init中的层, 定义到forward中的层无法显示, 并且层的先后顺序是init中的创建顺序

相关文章

网友评论

    本文标题:PyTorch 打印模型

    本文链接:https://www.haomeiwen.com/subject/wdewxqtx.html