From 5dff6c4e88aa3c15586515f25c4da636addd4f71 Mon Sep 17 00:00:00 2001 From: Liyuan Liu Date: Thu, 1 Nov 2018 21:11:55 +0100 Subject: [PATCH] Update main.py the current version of ``--cuda`` is store_false, maybe it would be better to change it --- main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 807df49..4861fe4 100644 --- a/main.py +++ b/main.py @@ -56,8 +56,8 @@ help='random seed') parser.add_argument('--nonmono', type=int, default=5, help='random seed') -parser.add_argument('--cuda', action='store_false', - help='use CUDA') +parser.add_argument('--nocuda', action='store_true', + help='not use CUDA') parser.add_argument('--log-interval', type=int, default=200, metavar='N', help='report interval') parser.add_argument('--save', type=str, default='EXP', @@ -104,8 +104,8 @@ def logging(s, print_=True, log_=True): np.random.seed(args.seed) torch.manual_seed(args.seed) if torch.cuda.is_available(): - if not args.cuda: - print("WARNING: You have a CUDA device, so you should probably run with --cuda") + if args.nocuda: + print("WARNING: You have a CUDA device, so you should probably run without --nocuda") else: torch.cuda.manual_seed_all(args.seed)