使用yolov5和yolov8训练都出现的一个错误:TypeError: expected Tensor as element 0 in argument 0, but got NoneType
最后发现是pytoch版本的原因,开始我的pytorch版本为>2.0,后面降到1.12就可以跑了
使用yolov5和yolov8训练出现的错误:TypeError: expected Tensor as element 0 in argument 0, but got NoneType
最新推荐文章于 2024-04-01 11:06:48 发布