RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘
问题:RuntimeError: “unfolded2d_copy” not implemented for ‘Half’在使用GPU训练完deepspeech2语音识别模型后,使用django部署模型,当输入传入到模型进行计算的时候,报出的错误,查了问题,模型传入的参数use_half=TRUE,就是利用fp16混合精度计算对CPU进行推理,使用fp16来加快速度。解决方式:pytorch conv cpu不支持fp16,所以只需要将use_half=False,这样就能进行计算了。...
原创
2021-01-13 14:28:41 ·
15334 阅读 ·
19 评论