admin 管理员组

文章数量: 1184232

今天在使用pytorch1.7训练模型，调用apex时出现了该问题：

Doubletensor是64位float类型，Floattensor是32位float类型，调试程序发现第二次迭代时，变量v和msd[k]的数据类型不同，如图所示：

因此在条件循环后添加程序：

                    if v.dtype != msd[k].dtype:
                        v = v.to(msd[k].dtype)

修改后如图：

解决问题！

本文标签： 解决问题 HANDLE types FloatTensor DoubleTensor

版权声明：本文标题：解决问题：Do not know how to handle these types to promote: {‘DoubleTensor‘, ‘FloatTensor‘} 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/b/1763929084a3279411.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。