pytorch打印自定义网络的每层的名称
发布日期:2021-07-01 04:36:55
浏览次数:2
分类:技术文章
本文共 1022 字,大约阅读时间需要 3 分钟。
pytorch打印自定义网络的每层的名称
import torchfrom torchvision import modelsfrom torchsummary import summaryfrom resnext_MulTask_clothes import resnext50_elasticdata_class=[8, 7]device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')# vgg = models.vgg16().to(device)model = resnext50_elastic(num_classes=data_class) # 原模型model = torch.nn.DataParallel(model).cuda() # 并行处理# 已训练好的模型的pth文件checkpoint = torch.load('06-resnext50_elastic_checkpoint.pth.tar')model.load_state_dict(checkpoint['state_dict'], strict=False) # 参数加载summary(model, (3, 224, 224))
参考连接:https://www.jianshu.com/p/97c626d33924
另:
打印resnet152网络的每层的名称import torchfrom torchvision import modelsfrom torchsummary import summaryfrom resnet_pretrained import resnet152device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')model = resnet152() # 原模型model = torch.nn.DataParallel(model).cuda() # 并行处理# 已训练好的模型的pth文件checkpoint = torch.load('resnet152-b121ed2d.pth')model.load_state_dict(checkpoint, strict=False) # 参数加载summary(model, (3, 224, 224))
转载地址:https://mymuli.blog.csdn.net/article/details/100834714 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
做的很好,不错不错
[***.243.131.199]2024年04月13日 05时39分09秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
对称加密整个过程
2021-07-04
java内存模型
2021-07-04
volatile关键字
2021-07-04
Servlet_快速入门
2021-07-04
Request_继承体系
2021-07-04
前端权限控制:获取用户信息接口构造数据
2021-07-04
七牛云存储:断点续传
2021-07-04
字节流复制文本文件【应用】
2021-07-04
私钥加密私钥解密
2021-07-04
锁的释放流程-ReentrantLock.unlock
2021-07-04
Java判断字符串是否为数字(浮点类型也包括)
2021-07-04
ubuntu opencv-python 安装很慢问题
2021-07-04
MySQL5.7版本修改了my.ini配置文件后mysql服务无法启动问题
2021-07-04
Azkaban体系结构
2019-05-01
机器学习之重头戏-特征预处理
2019-05-01
synchronized底层实现及锁的升级、降级
2019-05-01
PermGen space-永久区内存溢出
2019-05-01
Maven继承和聚合
2019-05-01