Skip to content

Latest commit

 

History

History
24 lines (14 loc) · 706 Bytes

README.md

File metadata and controls

24 lines (14 loc) · 706 Bytes

Graph Convolutional Network

This is a PyTorch implementation of the Graph Convolutional Network (Kipf and Welling, 2016).

Requirements

In order to install the required libraries it is recommended that you create a conda environment by running

conda env create -f environment.yaml

and before running any experiments activate the environment:

source activate gcn

Running

The file gcn.py accepts several arguments. For instance, to train a model with the default configuration on GPU 0 you may run the command:

python gcn.py --config_path default.yaml --gpu 0

When the --gpu argument is not used training is performed on the CPU.