few-shot-object-detection icon indicating copy to clipboard operation
few-shot-object-detection copied to clipboard

RuntimeError: CUDA error: device-side assert triggered

Open MrCrightH opened this issue 2 years ago • 2 comments

RuntimeError: CUDA error: device-side assert triggered

MrCrightH avatar Feb 27 '22 03:02 MrCrightH

I have the same question。Did you solved it?

zhang-blip avatar Sep 23 '22 02:09 zhang-blip

sampling.py deletes [device=negative.device] change: positive_numel = positive.numel() negative_numel = negative.numbel() perm1 = torch.randperm(positive_numel, device=positive.device)[:num_pos] perm2 = torch.randperm(negative_numel)[:num_neg] perm2 = perm2.to(negative.device)

lvwengmx avatar Nov 21 '22 06:11 lvwengmx