Errors occur when running 'bash train_attMPTI.sh'

[neymar-jr]

Errors are listed as below.

Traceback (most recent call last):
File "/home/liuxuanchen/Develop/few-shot-point-cloud/attMPTI/main.py", line 101, in
train(args)
File "/home/liuxuanchen/Develop/few-shot-point-cloud/attMPTI/runs/mpti_train.py", line 58, in train
loss, accuracy = MPTI.train(data)
File "/home/liuxuanchen/Develop/few-shot-point-cloud/attMPTI/models/mpti_learner.py", line 63, in train
query_logits, loss= self.model(support_x, support_y, query_x, query_y)
File "/home/liuxuanchen/anaconda3/envs/torch/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1194, in call_impl
return forward_call(*input, **kwargs)
File "/home/liuxuanchen/Develop/few-shot-point-cloud/attMPTI/models/mpti.py", line 107, in forward
A = self.calculateLocalConstrainedAffinity(node_feat, k=self.k_connect)
File "/home/liuxuanchen/Develop/few-shot-point-cloud/attMPTI/models/mpti.py", line 258, in calculateLocalConstrainedAffinity
A = A.scatter(1, I, knn_similarity)
RuntimeError: Expected index [4396, 200] to be smaller than self [4396, 4396] apart from dimension 1 and to be smaller size than src [4396, 192]

[mayur20169]

I am also getting exactly same error.

[suurajroshan]

I get a similar error too. Did anyone get a solution to this error?

[WeiTang-code]

It may be something wrong with faiss. I've managed to run the code successfully in another environment, but it gives me an error in a higher version of pytorch and cuda environment on a 30-series GPU. Now I get wrong knn_index.

---

Further troubleshooting revealed that it was not a faiss error, but a large number of "nan" in the tensor of node_feat, which comes from the prototype.

---

I have solved this problem. It comes from the function "pytorch.nn.functional.pairwise_distance". This function behaves differently in different versions of pytorch.

If you use pytorch of higher version, you can refer to my solution. Note that every use of the "pairwise_distance" function requires modification.

distances = F.pairwise_distance(feat[..., None].transpose(1,2), farthest_seeds[None, ...], p=2) # (n_points, n_prototypes)

[xin1106]

@WeiTang-code Your answer is very helpful.
I encountered an error while running the protonet.by :
Runtime Error: Expected target size [2,192], got [2,2048].
Have you encountered this problem before.
For the 'similarity=- F.pairise_distance (feature, prototype [None, None], p=2)**2'， how should I modify it?

[good-est]

Here are my adjustions on mpti.py. First, I replace the "distances = F.pairwise_distance(feat[..., None], farthest_seeds.transpose(0, 1)[None, ...], p=2) # (n_points, n_prototypes)“ in line 152 with “distances = F.pairwise_distance(feat[..., None].transpose(1,2), farthest_seeds[None, ...], p=2) # (n_points, n_prototypes)”. Second, I replece the "dist = F.pairwise_distance(node_feat[:,:,None], knn_feat.transpose(1,2), p=2)“ in line 251 with “dist = torch.cdist(node_feat.unsqueeze(1),knn_feat,p=2).squeeze(1)”.