Thank you for your contributions. Very interesting work!. I have two GPUs and I'm trying to run crosspoint pre-training for classification using:
Traceback (most recent call last):
File "train_crosspoint.py", line 258, in <module>
train(args, io)
File "train_crosspoint.py", line 100, in train
_, point_feats, _ = point_model(data)
File "/home/nas/anaconda3/envs/crosspoint/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/home/nas/Desktop/CrossPoint/models/dgcnn.py", line 95, in forward
x = get_graph_feature(x, k=self.k)
File "/home/nas/Desktop/CrossPoint/models/dgcnn.py", line 31, in get_graph_feature
idx = idx + idx_base
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cuda:1!