few-shot-object-detection
few-shot-object-detection copied to clipboard
RuntimeError: CUDA error: device-side assert triggered
RuntimeError: CUDA error: device-side assert triggered
I have the same question。Did you solved it?
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)