2019
DOI: 10.48550/arxiv.1903.02428
|View full text |Cite
Preprint
|
Sign up to set email alerts
|

Fast Graph Representation Learning with PyTorch Geometric

Abstract: We introduce PyTorch Geometric, a library for deep learning on irregularly structured input data such as graphs, point clouds and manifolds, built upon PyTorch. In addition to general graph data structures and processing methods, it contains a variety of recently published methods from the domains of relational learning and 3D data processing. PyTorch Geometric achieves high data throughput by leveraging sparse GPU acceleration, by providing dedicated CUDA kernels and by introducing efficient mini-batch handli… Show more

Help me understand this report

Search citation statements

Order By: Relevance

Paper Sections

Select...
3
1
1

Citation Types

0
528
0

Year Published

2021
2021
2023
2023

Publication Types

Select...
5
5

Relationship

0
10

Authors

Journals

citations
Cited by 478 publications
(528 citation statements)
references
References 19 publications
0
528
0
Order By: Relevance
“…All of the experiments were implemented using Python 3.8, Pytorch 1.7.1 and Pytorch Geometric 1.7.0 [60]. The code base that was used to perform these experiments is available 4 .…”
Section: Methodsmentioning
confidence: 99%
“…All of the experiments were implemented using Python 3.8, Pytorch 1.7.1 and Pytorch Geometric 1.7.0 [60]. The code base that was used to perform these experiments is available 4 .…”
Section: Methodsmentioning
confidence: 99%
“…The experiments were implemented using Python 3.8, Pytorch 1.7.1 and Pytorch Geometric 1.7.0 [17]. The code 2 was executed on a virtual machine running on 8 cores of an AMD EPYC 7551 32-Core Processor with an Nvidia V100 32GB GPU.…”
Section: Methodsmentioning
confidence: 99%
“…Our model is built upon Pytorch [28] and PyG (PyTorch Geometric) library [29]. We train MGAE for 200 epochs with Adam [30] optimizer and early stopping with a patience of 50 epochs.…”
Section: Methodsmentioning
confidence: 99%