debug YOLOX-style Transform with Rotation

models/detectors/yolox/loss.py

@@ -159,13 +159,6 @@ class Criterion(object):
                  self.loss_cls_weight * loss_cls + \
                  self.loss_box_weight * loss_box
 
-        loss_dict = dict(
-                loss_obj = loss_obj,
-                loss_cls = loss_cls,
-                loss_box = loss_box,
-                losses = losses
-        )
-
         # ------------------ Aux regression loss ------------------
         loss_box_aux = None
         if epoch >= (self.max_epoch - self.no_aug_epoch - 1):
@@ -201,7 +194,6 @@ class Criterion(object):
                     losses = losses
                     )
 
-
         return loss_dict