Nihal Gupta
Nihal Gupta
which version of torch scatter, are you using with pytorch 1.10
> Hello, thanks for reporting. > > From your log, it looks like the error happens in the middle of the training, after 400 batches, is that correct? If so,...
> Hello. I was able to complete one epoch of training using two GPUs without any errors, so I am not sure what is causing your problem. I have two...
i did set cuda 12.1, then tried still same issue as u asked printed the out_ch and slice_idx weight shape: torch.Size([256, 1, 3, 3]) x shape: torch.Size([10, 256, 44, 60])...