diff --git a/examples/GaAN/README.md b/examples/GaAN/README.md index a3b0981a63211fc2c2555d5942495ca849238e8c..c9c08f05f3ed7bbca03b41d268973caadebc4328 100644 --- a/examples/GaAN/README.md +++ b/examples/GaAN/README.md @@ -12,7 +12,7 @@ The ogbn-proteins dataset will be downloaded in directory ./dataset automaticall ## How to run ```bash -python train.py --lr 1e-2 --rc 0 --batch_size 1024 --epochs 100 +python train.py --lr 1e-2 --rc 0 --batch_size 1024 --epochs 100 --gpu_id 0 ``` or @@ -23,6 +23,7 @@ source main.sh ### Hyperparameters - use_gpu: whether to use gpu or not - mini_data: use a small dataset to test code +- gpu_id: the id of gpu - epochs: number of training epochs - lr: learning rate - rc: regularization coefficient