pytorch打印自定义网络的每层的名称
发布日期:2021-07-01 04:36:55
浏览次数:3
分类:技术文章
本文共 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 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!
发表评论
最新留言
路过,博主的博客真漂亮。。
[***.116.15.85]2024年04月20日 19时48分09秒
关于作者
喝酒易醉,品茶养心,人生如梦,品茶悟道,何以解忧?唯有杜康!
-- 愿君每日到此一游!
推荐文章
python import midi error
2019-05-07
MyBatis模糊查询的sql拼接
2019-05-07
CSS选择器
2019-05-07
eclipse安装aptana插件
2019-05-07
cocos2d-x auto-batching
2019-05-07
cocos-lua 使用plist文件
2019-05-07
gitlab迁移后一直提示413错误
2019-05-07
ElasticSearch查询语句概述
2019-05-07
springboot 数据连接池HikariPool
2019-05-07
基于角色得后台权限管理系统设计(一、表设计)
2019-05-07
maven得一些知识点
2019-05-07
优雅得运维应该使用软链接
2019-05-07
基于角色得后台权限管理系统设计(七、spring security 之请求鉴权(详解)
2019-05-07
基于角色得后台权限管理系统设计(八、spring security 之自定义前缀一)
2019-05-07
基于角色得后台权限管理系统设计(九、spring security 之自定义前缀二)
2019-05-07
基于角色得后台权限管理系统设计(十、spring security 之自定义前缀三)
2019-05-07
centos7.x安装最新版rabbitmq
2019-05-07
Spring Cloud Gateway
2019-05-07
很坑得spring cloud gateway 异常
2019-05-07