使用yolov5训练自己数据时报错：AttributeError: ‘NoneType‘ object has no attribute ‘_free_weak_ref‘

原创已于 2022-08-14 15:44:30 修改 · 3.4k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

于 2022-08-11 20:34:57 首次发布

本文记录了使用YoloV5 v6.1版本在Windows环境下训练自定义数据集时遇到的问题及解决方法。当设置--image-weights=True时出现错误，通过调整PyTorch版本或设置--workers=0解决问题，并分享了训练效果对比。

部署运行你感兴趣的模型镜像

最近使用yolov5v6.1训练自己的数据集，torch version == 1.11.0,本来一切正常，但是当我设置

--image-weights=Ture时报错：

Exception ignored in: <function StorageWeakRef.__del__ at 0x000001CFC9C2BB80>
Traceback (most recent call last):
  File "J:\anaconda\envs\pytorch_env_liu\lib\site-packages\torch\multiprocessing\reductions.py", line 36, in __del__
  File "J:\anaconda\envs\pytorch_env_liu\lib\site-packages\torch\storage.py", line 520, in _free_weak_ref
AttributeError: 'NoneType' object has no attribute '_free_weak_ref'

github上有这个issue的讨论，一种解决方法是降低pytorch版本小于1.11.0，我没有降低pytorch版本，设置了--workers==0,也即只使用一个主线程加载数据，不再报错。系统为windows系统。github关于该issue的讨论链接如下：github关于该issue的讨论

有兴趣的伙伴可以看看。

经过测试，发现开启了image_weights不仅训练速度变慢（worker==0），效果也比不适用变差，可能跟我的数据集有关系，大家可以在自己的数据集上试试。

您可能感兴趣的与本文相关的镜像

PyTorch 2.7

PyTorch

Cuda

PyTorch 是一个开源的 Python 机器学习库，基于 Torch 库，底层由 C++ 实现，应用于人工智能领域，如计算机视觉和自然语言处理