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,
解决方法
- 将use_half=False
- 或者将half()修改为float()
这样模型就能进行计算了;
我报错的修改:
希望这篇文章对你有用!
谢谢点赞评论!
以上是关于RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘(实测百分百有效)的主要内容,如果未能解决你的问题,请参考以下文章
RuntimeError: tf.placeholder() 与急切执行不兼容
RuntimeError:预期的标量类型 Double 但发现 Float
multiprocessing进程开发RuntimeError