initialize_model.py 文件源码

def get_parser():

    def dict_type(ss):
        return dict([map(str.strip, s.split(':'))
                     for s in ss.split(',')])

    parser = argparse.ArgumentParser()
    parser.add_argument('--proto', type=str)
    parser.add_argument('--ref-encs', type=dict_type,
                        help="Models to initialize encoders, \
                        eg. --ref-encs=fi:file1,de:file2")
    parser.add_argument('--ref-decs', type=dict_type,
                        help="Models to initialize decoders, \
                        eg. --ref-decs=en:file1,de:file2")
    parser.add_argument('--ref-att', type=str,
                        help="Model to initialize shared components")
    parser.add_argument('--ref-dec-embs', type=dict_type,
                        help="Models to initialize decoder embeddings, \
                        eg. --ref-dec-embs=en:file1,de:file2")
    parser.add_argument('--ref-enc-embs', type=dict_type,
                        help="Models to initialize encoder embeddings, \
                        eg. --ref-enc-embs=en:file1,de:file2")
    return parser