RuntimeError: CUDA error: device-side assert triggered

ucbdrive / few-shot-object-detection

Implementations of few-shot object detection benchmarks

Apache License 2.0

1.08k stars 225 forks source link

RuntimeError: CUDA error: device-side assert triggered #168

Open MrCrightH opened 2 years ago

MrCrightH commented 2 years ago

RuntimeError: CUDA error: device-side assert triggered

zhang-blip commented 1 year ago

I have the same question。Did you solved it？

lvwengmx commented 1 year ago

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)