search

Sharpiless / yolov5-knowledge-distillation

yolov5目标检测模型的知识蒸馏（基于响应的蒸馏）
GNU General Public License v3.0
91 stars 16 forks source link

KeyError: 'giou' #1

Open hyaihjq opened 3 years ago

hyaihjq commented 3 years ago

File "/home/hongyang/codebase/pytorch_code/yolov5-knowledge-distillation/utils/loss.py", line 57, in compute_distillation_output_loss t_lbox = h['giou'] h['dist'] KeyError: 'giou'

hyaihjq commented 3 years ago

代码跑不通

baigaoyi commented 2 years ago

看源码应该是作作者在hyp.scratch.yaml新定义了一些超参数，但不知道他的参数如何设定，项目代码里没有提供。老哥，你最后有没有继续跑这个项目呢？