RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）

Posted 醉公子～

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）相关的知识，希望对你有一定的参考价值。

报错

RuntimeError: "unfolded2d_copy" not implemented for 'Half'

原因

模型传入的参数use_half=true，就是利用fp16混合精度计算对CPU进行推理，使用fp16来加快速度 但是pytorch cpu不支持fp16，

解决方法

  1. 将use_half=False
  2. 或者将half()修改为float()

这样模型就能进行计算了；

我报错的修改：


希望这篇文章对你有用！

谢谢点赞评论！

以上是关于RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘（实测百分百有效）的主要内容，如果未能解决你的问题，请参考以下文章