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

teacher_model为什么要用t_model.train()? #5

Open asizk opened 3 years ago

asizk commented 3 years ago

teacher_model为什么要用t_model.train()?而不是t_model.eval() 我用了您的代码 在yolov5L上训练的map是71 在yolov5s上训练的map是68 在yolov5s上蒸馏出来的map只有35 请问您有什么好的想法吗,为什么会出现这种原因