Spaces:
Sleeping
Sleeping
Update python_utils/get_model.py
Browse files
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"
|