Explorar o código

debug loss of YOLOv2

yjh0410 %!s(int64=2) %!d(string=hai) anos
pai
achega
907c5993e7
Modificáronse 1 ficheiros con 0 adicións e 1 borrados
  1. 0 1
      models/yolov2/loss.py

+ 0 - 1
models/yolov2/loss.py

@@ -83,7 +83,6 @@ class Criterion(object):
         
         # obj loss
         gt_objectness[pos_masks] *= ious.clamp(0.)
-        print(gt_objectness[pos_masks])
         loss_obj = self.loss_objectness(pred_obj, gt_objectness)
         loss_obj = loss_obj.sum() / num_fgs