pyg-team/pytorch_geometric

Improve `NNConv` runtimes

Aperta

#602 aperta il 6 ago 2019

Vedi su GitHubGuida contributor
 (8 commenti) (0 reazioni) (0 assegnatari)Python (3514 fork)batch import
2 - Priority P2benchmarkfeaturehelp wantednn

Metriche repository

Star
 (19.985 stelle)
Metriche merge PR
 (Merge medio 35g 1h) (14 PR mergiate in 30 g)

Descrizione

🐛 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

Guida contributor

Direzione di ricerca
Profila il forward pass di NNConv per identificare i colli di bottiglia, in particolare nel calcolo dei messaggi dove viene utilizzato l'intero pseudo tensore.
Tech stack
pythonpytorch
Dominio
machine learning
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
PythonPyTorchgraph neural networks
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
50