会员
众包
新闻
博问
闪存
赞助商
HarmonyOS
Chat2DB
所有博客
当前博客
我的博客
我的园子
账号设置
会员中心
简洁模式
...
退出登录
注册
登录
不积跬步无以至千里
研究领域:深度学习,图像处理
联系方式:vladimirputin@foxmail.com
不必高看自己,也不必贬低自己
博客园
首页
新随笔
联系
管理
随笔分类 -
pytorch
上一页
1
2
3
torch加载模型出现***.path is a zip archieve的错误
摘要:原因: 该模型保存时使用的torch版本要高于此次加载时使用的torch版本 解决办法: 在高版本的torch中重新load该模型,在重新保存时,在save中的modepath后面加上:use_new_zipfile_serialization=False
阅读全文
posted @
2022-05-31 20:29
海_纳百川
阅读(52)
评论(0)
推荐(0)
torch.bmm()解读
摘要:函数作用 计算两个tensor的矩阵乘法,torch.bmm(a,b),tensor a 的size为(b,h,w),tensor b的size为(b,w,m) 也就是说两个tensor的第一维是相等的,然后第一个数组的第三维和第二个数组的第二维度要求一样,对于剩下的则不做要求,输出维度 (b,h,
阅读全文
posted @
2022-05-27 09:13
海_纳百川
阅读(945)
评论(0)
推荐(0)
上一页
1
2
3
公告
本站总访问量
次