RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘

问题:RuntimeError: “unfolded2d_copy” not implemented for ‘Half’
RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘_第1张图片在使用GPU训练完deepspeech2语音识别模型后,使用django部署模型,当输入传入到模型进行计算的时候,报出的错误,查了问题,模型传入的参数use_half=TRUE,就是利用fp16混合精度计算对CPU进行推理,使用fp16来加快速度。

解决方式:
pytorch conv cpu不支持fp16,所以只需要将use_half=False,或者将.half()修改成.float(),这样就能进行计算了。
RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘_第2张图片

你可能感兴趣的:(深度学习,深度学习)