config.py

import utils
import argparse
import json
def parse_args():
    desc ="Pytorch Adversarial Attack"
    parser = argparse.ArgumentParser(description=desc)
    parser.add_argument('--cfg_path', type=str, default='./config/train.json')
    return parser.parse_args()

def get_configs(args):
    with open(args.cfg_path, "r") as f:
        configs = json.load(f)
    arg_dict = vars(args)
    for key in arg_dict : 
        if key in configs:
            if arg_dict[key] is not None:
                configs[key] = arg_dict[key]
    configs = utils.ConfigMapper(configs)
    return configs