GHM_Detection
GHM_Detection copied to clipboard
Restriction on GHMC
there should be limitation on variable weights
, how to set it? Or which range is suitable?
I apply GHMC loss directly on rotated-yolo for conf loss calc, loss explosion appared resulted from too large weights:
tensor([ 3.82812, 3.82812, 512.00000, 47.28125, 3.82812, 3.82812, 47.28125, 3.82812, 3.82812, 3.82812, 47.28125, 47.28125, 3.82812, 3.82812, 47.28125, 165.00000, 3.82812, 3.82812, 47.28125, 165.00000, 3.82812, 3.82812, 3.82812, 3.82812, 3.82812, 928.00000, ..., ], device='cuda:0', dtype=torch.float16)