pyg-team/pytorch_geometric

Improve `NNConv` runtimes

Aberta

#602 aberto em 6 de ago. de 2019

Ver no GitHubGuia do colaborador
 (8 comentários) (0 reação) (0 responsável)Python (3.514 forks)batch import
2 - Priority P2benchmarkfeaturehelp wantednn

Métricas do repositório

Stars
 (19.985 estrelas)
Métricas de merge de PR
 (Mesclagem média 35d 1h) (14 fundiu PRs em 30d)

Description

🐛 Bug

I have benchmarked EEC vs https://github.com/mys007/ecc implementation. The code is 3-5 times slower in pytorch_geometric.

I think it is due to the entire pseudo tensor which is given to every message function. Maybe, only weight_j should be given there.

To Reproduce

Steps to reproduce the behavior:

Expected behavior

Environment

  • OS:
  • Python version:
  • PyTorch version:
  • CUDA/cuDNN version:
  • GCC version:
  • Any other relevant information:

Additional context

Guia do colaborador

Direção de pesquisa
Profile o forward pass do NNConv para identificar gargalos, especialmente no cálculo de mensagens onde o tensor pseudo completo é usado.
Pilha de tecnologia
pythonpytorch
Domain
machine learning
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
2
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Principalmente claro
Prerequisites
PythonPyTorchgraph neural networks
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
50