Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Refine criterion and matcher logging info #151

Merged
merged 1 commit into from Dec 1, 2022
Merged

Conversation

rentainhe
Copy link
Collaborator

@rentainhe rentainhe commented Dec 1, 2022

TODO

  • Refine logging for criterion and matcher to make it more readable in log follow Mask2Former

example

  • original logging in DINO
(criterion): DINOCriterion(
  (matcher): HungarianMatcher()
)
  • modified version
(criterion): Criterion DINOCriterion
    matcher: Matcher HungarianMatcher
        cost_class: 2.0
        cost_bbox: 5.0
        cost_giou: 2.0
        cost_class_type: focal_loss_cost
        focal cost alpha: 0.25
        focal cost gamma: 2.0
    losses: ['class', 'boxes']
    loss_class_type: focal_loss
    weight_dict: {'loss_class': 1, 'loss_bbox': 5.0, 'loss_giou': 2.0, 'loss_class_dn': 1, 'loss_bbox_dn': 5.0, 'loss_giou_dn': 2.0, 'loss_class_enc': 1, 'loss_bbox_enc': 5.0, 'loss_giou_enc': 2.0, 'loss_class_dn_enc': 1, 'loss_bbox_dn_enc': 5.0, 'loss_giou_dn_enc': 2.0, 'loss_class_0': 1, 'loss_bbox_0': 5.0, 'loss_giou_0': 2.0, 'loss_class_dn_0': 1, 'loss_bbox_dn_0': 5.0, 'loss_giou_dn_0': 2.0, 'loss_class_1': 1, 'loss_bbox_1': 5.0, 'loss_giou_1': 2.0, 'loss_class_dn_1': 1, 'loss_bbox_dn_1': 5.0, 'loss_giou_dn_1': 2.0, 'loss_class_2': 1, 'loss_bbox_2': 5.0, 'loss_giou_2': 2.0, 'loss_class_dn_2': 1, 'loss_bbox_dn_2': 5.0, 'loss_giou_dn_2': 2.0, 'loss_class_3': 1, 'loss_bbox_3': 5.0, 'loss_giou_3': 2.0, 'loss_class_dn_3': 1, 'loss_bbox_dn_3': 5.0, 'loss_giou_dn_3': 2.0, 'loss_class_4': 1, 'loss_bbox_4': 5.0, 'loss_giou_4': 2.0, 'loss_class_dn_4': 1, 'loss_bbox_dn_4': 5.0, 'loss_giou_dn_4': 2.0}
    num_classes: 80
    eos_coef: None
    focal loss alpha: 0.25
    focal loss gamma: 2.0

@rentainhe rentainhe merged commit dfbc4d7 into main Dec 1, 2022
@rentainhe rentainhe deleted the refine_logging branch December 1, 2022 03:48
Lontoone pushed a commit to Lontoone/detrex that referenced this pull request Jan 8, 2024
Co-authored-by: ntianhe ren <rentianhe@dgx061.scc.idea>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant