pytorch 查看网络结构、参数

from torchsummary import summary
import torchvision.models as models
model = models.resnet18()
summary(model, input_size=(3,224,224), batch_size=-1)

直接修改网络层

model = resnet18(pretrained=False)
model.conv1 = nn.Conv2d(1, 64, kernel_size=(7, 7), stride=(2, 2), padding=(3, 3), bias=False)
发布了23 篇原创文章 · 获赞 2 · 访问量 1564
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览