Open SyGoing opened 5 years ago
我用你的代码在yolov2稀疏化训练时(yolov2 608输入,两类目标), 1)报错提示: 损失非常大,然后崩了,梯度爆炸。还有就是utils.py里的build_targets 提示index=19 in dimension 3 超过边界。 2)源码yolomodel.py中 self.losses["recall"] /= 3 self.losses["precision"] /= 3 为什么要除于3,是因为yolov3有三个输出层么? 那么yolov2的话是不是需要修改呢
我用你的代码在yolov2稀疏化训练时(yolov2 608输入,两类目标), 1)报错提示: 损失非常大,然后崩了,梯度爆炸。还有就是utils.py里的build_targets 提示index=19 in dimension 3 超过边界。 2)源码yolomodel.py中 self.losses["recall"] /= 3 self.losses["precision"] /= 3 为什么要除于3,是因为yolov3有三个输出层么? 那么yolov2的话是不是需要修改呢