Nlp 如何保存和加载simpletransformer模型？_Nlp_Question Answering_Simpletransformers

Nlp 如何保存和加载simpletransformer模型？

nlp

Nlp 如何保存和加载simpletransformer模型？,nlp,question-answering,simpletransformers,Nlp,Question Answering,Simpletransformers,我已经编写了一个脚本，其中我使用simpletranformer框架来训练问答模型。我现在想做预测，但正如我想象的那样，我必须从训练过的模型中保存权重，并将它们加载到我的预测脚本中，但我还无法解决如何做到这一点。下面的代码片段是我如何训练模型的 model = QuestionAnsweringModel("distilbert", "distilbert-base-uncased-distilled-s

我已经编写了一个脚本，其中我使用simpletranformer框架来训练问答模型。我现在想做预测，但正如我想象的那样，我必须从训练过的模型中保存权重，并将它们加载到我的预测脚本中，但我还无法解决如何做到这一点。下面的代码片段是我如何训练模型的

model = QuestionAnsweringModel("distilbert",
                               "distilbert-base-uncased-distilled-squad",
                               args= {'reprocess_input_data': True,
                                     'overwrite_output_dir': True,
                                     'learning_rate': 5e-5,
                                     'num_train_epochs': 3,
                                     'max_seq_length': 192,
                                     'doc_stride': 64,
                                     'fp16': False,
                                    },
                               use_cuda= cuda_available)


model.train_model(config.QA_TRAIN)