您好，如何将自己训练好的参数加载到模型中，完成事件抽取任务？

THU-KEG / OmniEvent

A comprehensive, unified and modular event extraction toolkit.

https://omnievent.readthedocs.io/

MIT License

341 stars 33 forks source link

Closed DangQD closed 2 years ago

yaof20 commented 2 years ago

您好，

关于参数加载：您可以选择checkpoint的路径，通过from_pretrained方法(如有)加载参数；或者参考pytorch的load_state_dict等方法加载参数。

关于推理：根据模型的forward的定义方式完成您所需要的任务。