HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

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

Browse source
This commit is contained in:
Laughing 2023-06-19 22:09:06 +08:00 • committed by GitHub
parent 58bccb1a9f
commit eda7758521
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 1 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
ultralytics/vit/utils/ops.py
View file

@ -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: