Spaces:
Running
Running
jiang
commited on
Commit
•
cbe5e70
1
Parent(s):
56bbba2
update
Browse files
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 =
|
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"}
|