Update args.py

args.py

@@ -71,7 +71,7 @@ def get_args_parser():
                         help='path where to save')
     parser.add_argument('--output_dir', default='',
                         help='path where to save, empty for no saving')
-    parser.add_argument('--device', default='cuda',
+    parser.add_argument('--device', default='cpu',
                         help='device to use for training / testing')
     parser.add_argument('--seed', default=0, type=int)
     parser.add_argument('--num_workers', default=8, type=int)