search

997261095 / bert-kbqa

基于bert的kbqa系统
148 stars 37 forks source link

SIM_main.py中--pre_train_model需要加载的预训练模型在哪里? #7

Open ttweixiao9999 opened 2 years ago

ttweixiao9999 commented 2 years ago

你好，我在运行SIM_main.py文件的时候，在参数设置中遇到了问题，我参数设置如下，但是不知道pre_train_model文件是哪个？而且除了pre_train_model不知道写哪个之外想请问下除此之外的配置对吗？希望能得到帮助，谢谢。

    parser.add_argument("--data_dir", default="input/data/ner_data", type=str, required=False,
                        help="数据文件目录，因当有train.txt dev.txt")
    parser.add_argument("--vob_file", default="input/config/bert-base-chinese-vocab.txt", type=str, required=False,
                        help="词表文件")
    parser.add_argument("--model_config", default="input/config/bert-base-chinese-config.json", type=str, required=False,
                        help="模型配置文件json文件")
    parser.add_argument("--pre_train_model", default=None, type=str, required=False, help="预训练的模型文件，参数矩阵。如果存在就加载")
    parser.add_argument("--output_dir", default="output", type=str, required=False, help="输出结果的文件")