Traceback (most recent call last):
File "E:/yolov5-master/train.py", line 547, in <module>
train(hyp, opt, device, tb_writer)
File "E:/yolov5-master/train.py", line 93, in train
state_dict = ckpt['model'].float().state_dict() # to FP32
File "F:\ANACONDA\envs\torch107\lib\site-packages\torch\nn\modules\module.py", line 490, in float
return self._apply(lambda t: t.float() if t.is_floating_point() else t)
File "F:\ANACONDA\envs\torch107\lib\site-packages\torch\nn\modules\module.py", line 359, in _apply
module._apply(fn)
File "F:\ANACONDA\envs\torch107\lib\site-packages\torch\nn\modules\module.py", line 359, in _apply
module._apply(fn)
File "F:\ANACONDA\envs\torch107\lib\site-packages\torch\nn\modules\module.py", line 359, in _apply
module._apply(fn)
[Previous line repeated 1 more time]
File "F:\ANACONDA\envs\torch107\lib\site-packages\torch\nn\modules\module.py", line 381, in _apply
param_applied = fn(param)
File "F:\ANACONDA\envs\torch107\lib\site-packages\torch\nn\modules\module.py", line 490, in <lambda>
return self._apply(lambda t: t.float() if t.is_floating_point() else t)
有没有大佬知道怎样解决，非常感谢！