test_v2.py 567 B

12345678910111213141516171819
  1. import os
  2. current_dir = os.path.dirname(os.path.abspath(__file__))
  3. parent_dir = os.path.dirname(current_dir)
  4. if parent_dir not in os.sys.path:
  5. os.sys.path.insert(0, parent_dir)
  6. from lib.utils.config import load_yaml_config
  7. for path in [
  8. "configs/segmentation/train_sup_us_template.yaml",
  9. "configs/segmentation/us_exp_sup_busi.yaml",
  10. "configs/segmentation/us_exp_sup_busi_ablation.yaml",
  11. ]:
  12. cfg = load_yaml_config(path)
  13. print(path)
  14. print("dataset:", cfg["dataset"]["dataset_name"])
  15. print("model keys:", sorted(cfg["model"].keys()))