/Benchmark

Single machine scalability test for DGL

Primary LanguagePython

Environment

  • Model: GIN
  • Dataset : Ogbg-molhiv / ogbg-molpcba
  • Number of layers: 3
  • Metric: Training time per epoch.
  • Machine: Aws EC2 g4dn.metal(8 T4 GPUs)
  • DGL version: 1.1
  • PyG version: 2.2.0
  • OS: ubuntu

Run

DGL

python run.py

PyG

python run.py --backend pyg