search

LeeSureman / Flat-Lattice-Transformer

code for ACL 2020 paper: FLAT: Chinese NER Using Flat-Lattice Transformer
1k stars 178 forks source link

模型保存与预测 #59

Open Endlesspace opened 3 years ago

Endlesspace commented 3 years ago

请问目前的代码中包含模型的保存和使用模型预测的部分吗？

LeeSureman commented 3 years ago

没有

okcd00 commented 3 years ago

保存模型的话可以考虑给 flat_main.py 里面的 Trainer 初始化的时候加个 save_path 传参，取值为模型存放的目录

Shajiu commented 2 years ago

保存模型的话可以考虑给 flat_main.py 里面的 Trainer 初始化的时候加个 save_path 传参，取值为模型存放的目录

请问具体是怎么加载的呢？有啥技巧么？