Sharpiless / yolov5-distillation-5.0

yolov5 5.0 version distillation || yolov5 5.0版本知识蒸馏,yolov5l >> yolov5s
GNU General Public License v3.0
153 stars 27 forks source link

train.py的一个新错误 #10

Open Qingmeiza opened 2 years ago

Qingmeiza commented 2 years ago

Traceback (most recent call last): File "C:/Users/qingmei/Desktop/yolov5-distillation-5.0-main/train.py", line 674, in train(hyp, opt, device, tb_writer) File "C:/Users/qingmei/Desktop/yolov5-distillation-5.0-main/train.py", line 254, in train t_model) else t_model.model[-1] UnboundLocalError: local variable 't_model' referenced before assignment

不知道该修改哪部分,有无大佬救一救(孩子跪求)

WangYafei0904 commented 2 years ago

你解决了么

sduaa commented 11 months ago

我也碰到了

1781686573 commented 9 months ago

请问解决了吗大佬,我也碰到这个问题了

sduaa commented 9 months ago

解决了,在命令行 python train.py --distill 就好了,应该设定好这个模式,好像是这样吧