jatamura commited on
Commit
f0542d9
·
verified ·
1 Parent(s): 659a4c2

Update python_utils/get_model.py

Browse files
Files changed (1) hide show
  1. python_utils/get_model.py +1 -2
python_utils/get_model.py CHANGED
@@ -24,8 +24,7 @@ def load_model():
24
 
25
  ## define relevant parameters
26
  cfg = get_cfg()
27
- cfg.merge_from_file(
28
- model_zoo.get_config_file("./configs/test_model_config.yaml"))
29
  cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("./configs/test_model_config.yaml")
30
  if not torch.cuda.is_available():
31
  cfg.MODEL.DEVICE = "cpu"
 
24
 
25
  ## define relevant parameters
26
  cfg = get_cfg()
27
+ cfg.merge_from_file("./configs/test_model_config.yaml")
 
28
  cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("./configs/test_model_config.yaml")
29
  if not torch.cuda.is_available():
30
  cfg.MODEL.DEVICE = "cpu"