jiang commited on
Commit
cbe5e70
1 Parent(s): 56bbba2
Files changed (1) hide show
  1. demo.py +1 -1
demo.py CHANGED
@@ -16,7 +16,7 @@ tasks.register_task('refcoco', RefcocoTask)
16
  # turn on cuda if GPU is available
17
  use_cuda = torch.cuda.is_available()
18
  # use fp16 only when GPU is available
19
- use_fp16 = True
20
 
21
  # Load pretrained ckpt & config
22
  overrides={"bpe_dir":"utils/BPE"}
 
16
  # turn on cuda if GPU is available
17
  use_cuda = torch.cuda.is_available()
18
  # use fp16 only when GPU is available
19
+ use_fp16 = use_cuda
20
 
21
  # Load pretrained ckpt & config
22
  overrides={"bpe_dir":"utils/BPE"}