随笔分类 -  pytorch

摘要:原因: 该模型保存时使用的torch版本要高于此次加载时使用的torch版本 解决办法: 在高版本的torch中重新load该模型,在重新保存时,在save中的modepath后面加上:use_new_zipfile_serialization=False 阅读全文
posted @ 2022-05-31 20:29 海_纳百川 阅读(52) 评论(0) 推荐(0)
摘要:函数作用 计算两个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)

本站总访问量