Факел-геометрическая функция softmax выбрасывает сегфо - PullRequest
0 голосов
/ 07 октября 2019

Я пытаюсь запустить некоторый код нейронной сети графа, который я нашел на github, но он выдает «Ошибка сегментации» в функции torch_geometric.utils softmax (). Как я могу это исправить?

Я пытался позвонить: softmax(alpha, edge_index[0], num_nodes), где:

  • alpha is torch.Size ([1266, 4])

  • edge_index [0] - torch.Size ([1266])

  • num_nodes - 561.

Iвзяла пик внутри torch_geometric.utils.softmax () и обнаружила, что код неисправен в строке, имеющей дело с torch_scatter.scatter_max(), но кроме этого я не уверен, как решить проблему.

Помощь оценена,спасибо! * * 1023

...