| 12345678910111213141516171819 |
- import os
- current_dir = os.path.dirname(os.path.abspath(__file__))
- parent_dir = os.path.dirname(current_dir)
- if parent_dir not in os.sys.path:
- os.sys.path.insert(0, parent_dir)
- from lib.utils.config import load_yaml_config
- for path in [
- "configs/segmentation/train_sup_us_template.yaml",
- "configs/segmentation/us_exp_sup_busi.yaml",
- "configs/segmentation/us_exp_sup_busi_ablation.yaml",
- ]:
- cfg = load_yaml_config(path)
- print(path)
- print("dataset:", cfg["dataset"]["dataset_name"])
- print("model keys:", sorted(cfg["model"].keys()))
|