From 7c69ddd9de4c0dc55ea09540051b7675a366b862 Mon Sep 17 00:00:00 2001
From: WangWenjin <30342227+WenjinW@users.noreply.github.com>
Date: Thu, 18 Jun 2020 15:55:33 +0800
Subject: [PATCH] update the description of hyper-parameter gpu_id

---
 examples/GaAN/README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/GaAN/README.md b/examples/GaAN/README.md
index a3b0981..c9c08f0 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
-- 
GitLab