pytorch+yolov5+deepsort训练报错RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘

原因:CPU训练,不支持半精度数

解决方法:将检测模块中的.half()全部改成.float()(计算时请保证数据类型一致)

pytorch+yolov5+deepsort训练报错RuntimeError: “unfolded2d_copy“ not implemented for ‘Half‘_第1张图片

 

参考资料:https://blog.csdn.net/hxx123520/article/details/112562930

你可能感兴趣的:(烂笔头,python,pytorch)