args = parser.parse_args() return args if __name__ == '__main__': args = parse_args() cfg_file = "configs/{}/{}.yml".format(args.DATASET, args.MODEL) with open(cfg_file, 'r') as f: yaml_dict = yaml.load(f) __C = CfgLoader(yaml_dict['MODEL_USE']).load() args = __C.str_to_bool(args) args_dict = __C.parse_to_dict(args) args_dict = {**yaml_dict, **args_dict} __C.add_args(args_dict) __C.proc() print('Hyper Parameters:') print(__C) execution = Execution(__C) execution.run(__C.RUN_MODE)
dest='WANDB', choices=['True', 'False'], help='True: Turn on WANDB logging,' 'False: Turn off WANDB logging', type=str) args = parser.parse_args() return args if __name__ == '__main__': args = parse_args() cfg_file = "configs/{}/{}.yml".format(args.DATASET, args.MODEL) with open(cfg_file, 'r') as f: yaml_dict = yaml.load(f) __C = CfgLoader(yaml_dict['MODEL_USE']).load() args = __C.str_to_bool(args) args_dict = __C.parse_to_dict(args) args_dict = {**yaml_dict, **args_dict} __C.add_args(args_dict) __C.proc() print('Hyper Parameters:') print(__C) execution = Execution(__C, args_dict) execution.run(__C.RUN_MODE)