Project-I/utils/args_reader.py

13 lines
317 B
Python

import argparse
import yaml
def config_loader():
parser = argparse.ArgumentParser()
parser.add_argument("--config", type=str, default="./config/DDGCRN_config.yaml")
config_path = parser.parse_args().config
with open(config_path, "r") as f:
config = yaml.safe_load(f)
return config