SsisyphusTao / Object-Detection-Knowledge-Distillation

An Object Detection Knowledge Distillation framework powered by pytorch, now having SSD and yolov5.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

使用retinaface-resnet50对rentinaface-mobile0.25进行蒸馏训练时,训练几个批次后loss_reg先变为Inf,其他损失函数变为NAN,想请问一下,一般是什么原因?不知道应该从哪方面改进

zhulingling1995 opened this issue · comments

使用retinaface-resnet50对rentinaface-mobile0.25进行蒸馏训练时,训练几个批次后loss_reg先变为Inf,其他损失函数变为NAN,想请问一下,一般是什么原因?不知道应该从哪方面改进