Zhihao Duan

Previously I had that problem too. For me, I just change the line 14 in CC.py to `self.CCN = torch.nn.DataParallel(self.CCN, device_ids=[0]).cuda()` Then everything is working. For details you may refer...

I successfully compiled on Windows 10, CUDA 11.1 (RTX3090), and PyTorch 1.7. Thank you so much!