Merge pull request #92 from WenjinW/master

Modify the setting of gpu in GaAN example

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

examples/GaAN/train.py

@@ -35,7 +35,7 @@ if __name__ == "__main__":
                        help="use a small dataset to test the code")
     parser.add_argument("--use_gpu", type=bool, choices=[True, False], default=True,
                        help="use gpu")
-    parser.add_argument("--gpu_id", type=int, default=4,
+    parser.add_argument("--gpu_id", type=int, default=0,
                        help="the id of gpu")
     parser.add_argument("--exp_id", type=int, default=0,
                        help="the id of experiment")