未验证 提交 3234641a 编写于 作者: W WangWenjin 提交者: GitHub

Modify the default gpu id

上级 51634002
...@@ -35,7 +35,7 @@ if __name__ == "__main__": ...@@ -35,7 +35,7 @@ if __name__ == "__main__":
help="use a small dataset to test the code") help="use a small dataset to test the code")
parser.add_argument("--use_gpu", type=bool, choices=[True, False], default=True, parser.add_argument("--use_gpu", type=bool, choices=[True, False], default=True,
help="use gpu") 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") help="the id of gpu")
parser.add_argument("--exp_id", type=int, default=0, parser.add_argument("--exp_id", type=int, default=0,
help="the id of experiment") help="the id of experiment")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册