diff --git a/examples/GaAN/train.py b/examples/GaAN/train.py index 6a8f905e158b21222bef67cd0dc4c6232113c42c..3c7a33929b05641f5943ff2d210dd4fd0a8b19e9 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")