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‘(实测百分百有效)的主要内容,如果未能解决你的问题,请参考以下文章

RuntimeError: tf.placeholder() 与急切执行不兼容

RuntimeError:预期的标量类型 Double 但发现 Float

RuntimeError:创建新内容类型时出错

multiprocessing进程开发RuntimeError

Python Asyncio - RuntimeError:无法关闭正在运行的事件循环

第二次运行 pyqtgraph 时发生 RuntimeError