update

examples/GaAN/main.sh

-python3 train.py --epochs 100 --lr 1e-2 --rc 0 --batch_size 1024 --gpu_id 0 --exp_id 0
\ No newline at end of file
+python3 train.py --epochs 100 --lr 1e-2 --rc 0 --batch_size 1024 --exp_id 0
\ No newline at end of file

examples/GaAN/train.py

@@ -82,7 +82,7 @@ if __name__ == "__main__":
     batch_size = args.batch_size
     sample_workers = 1
                         
-    place = fluid.CUDAPlace(0) if args.use_gpu else fluid.CPUPlace()           
+    place = fluid.CUDAPlace(args.gpu_id) if args.use_gpu else fluid.CPUPlace()           
     train_program = fluid.Program()
     startup_program = fluid.Program()