YuAnthony commited on
Commit
b43c0b1
·
1 Parent(s): 6e41f12

Update infer_contrast.py

Browse files
Files changed (1) hide show
  1. infer_contrast.py +1 -1
infer_contrast.py CHANGED
@@ -16,7 +16,7 @@ add_arg('model_path', str, 'models_large/resnet34.pth', '预测模
16
  args =parser.parse_known_args()[0]
17
 
18
  print_arguments(args)
19
-
20
  device = torch.device("cpu")
21
 
22
  # 加载模型
 
16
  args =parser.parse_known_args()[0]
17
 
18
  print_arguments(args)
19
+ print(torch.cuda.is_available())
20
  device = torch.device("cpu")
21
 
22
  # 加载模型