[RTDETR]fix "Grad strides do not match bucket view strides" when training with DDP (#3255)

2023-06-19 22:09:06 +08:00
parent 58bccb1a9f
commit eda7758521
2 changed files with 1 additions and 4 deletions
--- a/ultralytics/vit/utils/ops.py
+++ b/ultralytics/vit/utils/ops.py
@ -31,9 +31,6 @@ class HungarianMatcher(nn.Module):
        _cost_mask(bs, num_gts, masks=None, gt_mask=None): Computes the mask cost and dice cost if masks are predicted.
    """

-    class HungarianMatcher(nn.Module):
-        ...
-
    def __init__(self, cost_gain=None, use_fl=True, with_mask=False, num_sample_points=12544, alpha=0.25, gamma=2.0):
        super().__init__()
        if cost_gain is None: