会员
周边
众包
新闻
博问
闪存
赞助商
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
kenvision
博客园
首页
新随笔
联系
订阅
管理
2021年12月18日
TorchSummary无法载入Dict数据类型解决方法
摘要: 前言 torchsummary是一个比较不错的评价网络数据结构的包,但是目前torchsummary只支持网络输入和输出为torch.Tensor类型的网络,在对一些较为复杂的网络模型中,载入的可能并不一定为tensor类型,也有可能是list或者dict类型的数据。 为了可以支持自己的网络结构,为
阅读全文
posted @ 2021-12-18 14:28 kenvision
阅读(873)
评论(0)
推荐(0)
编辑
公告