Spaces:

zhigangjiang
/

3D-Room-Layout-Estimation_LGT-Net

Running

zhigangjiang commited on Nov 1, 2023

Commit

19f6f08

•

1 Parent(s): d589b70

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ def get_model(args):
     config = get_config(args)
     down_ckpt(args.cfg, config.CKPT.DIR)
     if ('cuda' in args.device or 'cuda' in config.TRAIN.DEVICE) and not torch.cuda.is_available():
-        logger.info(f'The {args.device} is not available, will use cpu ...')
         config.defrost()
         args.device = "cpu"
         config.TRAIN.DEVICE = "cpu"

     config = get_config(args)
     down_ckpt(args.cfg, config.CKPT.DIR)
     if ('cuda' in args.device or 'cuda' in config.TRAIN.DEVICE) and not torch.cuda.is_available():
+        logger.info(f'The {args.device} is not available, will use cpu...')
         config.defrost()
         args.device = "cpu"
         config.TRAIN.DEVICE = "cpu"