"official/projects/movinet/train.py" did not exist on "5b9f6e2f31f250bdb0352a53f79147ed84c9c73f"
README.md 299 Bytes
Newer Older
rusty1s's avatar
rusty1s committed
1
2
3
4
5
6
7
8
9
10
# Benchmark on Large-Scale Graphs

```
python main.py model=pna dataset=flickr root=/tmp/datasets device=0 log_every=1
```

You can choose between the following models and datasets:

* **Models:** `gcn`, `gcn2`, `pna`, `pna_jk`
* **Datasets:** `reddit`, `ppi`, `flickr`, `yelp`, `arxiv`, `products`