Update extract_feature_print.py
Browse files- extract_feature_print.py +1 -1
extract_feature_print.py
CHANGED
|
@@ -177,7 +177,7 @@ def main():
|
|
| 177 |
sys.exit(1)
|
| 178 |
|
| 179 |
# Load model
|
| 180 |
-
model_config = model_configs.get(model_name, model_configs[
|
| 181 |
model_dict = model_config["load_model"](model_path, config.device, config.is_half)
|
| 182 |
model = model_dict["model"]
|
| 183 |
additional_configs = model_dict.get("saved_cfg")
|
|
|
|
| 177 |
sys.exit(1)
|
| 178 |
|
| 179 |
# Load model
|
| 180 |
+
model_config = model_configs.get(model_name, model_configs[model_name])
|
| 181 |
model_dict = model_config["load_model"](model_path, config.device, config.is_half)
|
| 182 |
model = model_dict["model"]
|
| 183 |
additional_configs = model_dict.get("saved_cfg")
|