Skip to content

Commit

Permalink
Update the igbh readme of single-node multi-GPU training (#63)
Browse files Browse the repository at this point in the history
Update the igbh example readme
  • Loading branch information
LiSu authored Jul 24, 2023
1 parent ed7107a commit fc6c833
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions examples/igbh/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -30,11 +30,16 @@ in the training script below.

Note that in `dataset.py`, we have converted the graph into an undirected graph.

## 2. Single node and single GPU training:
## 2. Single node training:
```
python train_rgnn.py --model='rgat' --dataset_size='tiny' --num_classes=19
```
The script uses GPU default, please add `--cpu_mode` if you want to use CPU only.
The script uses a single GPU, please add `--cpu_mode` if you want to use CPU only.

To train the model using multiple GPUs:
```
CUDA_VISIBLE_DEVICES=0,1 python train_rgnn_multi_gpu.py --model='rgat' --dataset_size='tiny' --num_classes=19
```

## 3. Distributed (multi nodes) examples

Expand Down

0 comments on commit fc6c833

Please sign in to comment.