From 3234641a152ea46eaa142164f68a7a577b4b8426 Mon Sep 17 00:00:00 2001
From: WangWenjin <30342227+WenjinW@users.noreply.github.com>
Date: Thu, 18 Jun 2020 15:44:57 +0800
Subject: [PATCH] Modify the default gpu id

---
 examples/GaAN/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/GaAN/train.py b/examples/GaAN/train.py
index 6a8f905..3c7a339 100644
--- a/examples/GaAN/train.py
+++ b/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")
-- 
GitLab