参数解析看以下代码即可领悟:
parser = argparse.ArgumentParser()
parser.add_argument('--sparse', action='store_true', help='GAT with sparse version or not.')
parser.add_argument('--seed', type=int, default=72, help='Random seed.')
parser.add_argument('--epochs', type=int, default=10000, help='Number of epochs to train.')
args = parser.parse_args()
print(args.sparse)
print(args.seed)
print(args.epochs)
打印如下:
False
72
10000
读入yaml文件:
cfg.merge_from_list(yaml_file),则cfg中会存有yaml_file中的各种key-value,可用cfg.key获取对应的value。