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

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

--image-weights=Ture时报错：

Exception ignored in: 
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），效果也比不适用变差，可能跟我的数据集有关系，大家可以在自己的数据集上试试。