refine criterion and matcher logging (#151)

Co-authored-by: ntianhe ren <[email protected]>

Author: rentainhe

detrex/modeling/criterion/criterion.py

@@ -245,3 +245,20 @@ def forward(self, outputs, targets, return_indices=False):
             return losses, indices_list
 
         return losses
+
+
+    def __repr__(self):
+        head = "Criterion " + self.__class__.__name__
+        body = [
+            "matcher: {}".format(self.matcher.__repr__(_repr_indent=8)),
+            "losses: {}".format(self.losses),
+            "loss_class_type: {}".format(self.loss_class_type),
+            "weight_dict: {}".format(self.weight_dict),
+            "num_classes: {}".format(self.num_classes),
+            "eos_coef: {}".format(self.eos_coef),
+            "focal loss alpha: {}".format(self.alpha),
+            "focal loss gamma: {}".format(self.gamma),
+        ]
+        _repr_indent = 4
+        lines = [head] + [" " * _repr_indent + line for line in body]
+        return "\n".join(lines)

detrex/modeling/matcher/matcher.py

@@ -149,3 +149,16 @@ def forward(self, outputs, targets):
             (torch.as_tensor(i, dtype=torch.int64), torch.as_tensor(j, dtype=torch.int64))
             for i, j in indices
         ]
+
+    def __repr__(self, _repr_indent=4):
+        head = "Matcher " + self.__class__.__name__
+        body = [
+            "cost_class: {}".format(self.cost_class),
+            "cost_bbox: {}".format(self.cost_bbox),
+            "cost_giou: {}".format(self.cost_giou),
+            "cost_class_type: {}".format(self.cost_class_type),
+            "focal cost alpha: {}".format(self.alpha),
+            "focal cost gamma: {}".format(self.gamma),
+        ]
+        lines = [head] + [" " * _repr_indent + line for line in body]
+        return "\n".join(lines)