zhigangjiang commited on
Commit
19f6f08
1 Parent(s): d589b70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"