问题: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(),这样就能进行计算了。