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

解决方式:
pytorch conv cpu不支持fp16,所以只需要将use_half=False,或者将.half()修改成.float(),这样就能进行计算了。
在这里插入图片描述

Logo

CSDN联合极客时间,共同打造面向开发者的精品内容学习社区,助力成长!

更多推荐