zhigangjiang
commited on
Commit
•
19f6f08
1
Parent(s):
d589b70
Update app.py
Browse files
app.py
CHANGED
@@ -77,7 +77,7 @@ def get_model(args):
|
|
77 |
config = get_config(args)
|
78 |
down_ckpt(args.cfg, config.CKPT.DIR)
|
79 |
if ('cuda' in args.device or 'cuda' in config.TRAIN.DEVICE) and not torch.cuda.is_available():
|
80 |
-
logger.info(f'The {args.device} is not available, will use cpu
|
81 |
config.defrost()
|
82 |
args.device = "cpu"
|
83 |
config.TRAIN.DEVICE = "cpu"
|
|
|
77 |
config = get_config(args)
|
78 |
down_ckpt(args.cfg, config.CKPT.DIR)
|
79 |
if ('cuda' in args.device or 'cuda' in config.TRAIN.DEVICE) and not torch.cuda.is_available():
|
80 |
+
logger.info(f'The {args.device} is not available, will use cpu...')
|
81 |
config.defrost()
|
82 |
args.device = "cpu"
|
83 |
config.TRAIN.DEVICE = "cpu"
|