Implement proper saving/loading of BertMesh to/from local

wellcometrust / grants_tagger

Tag grants with MeSH and other tags

MIT License

14 stars 4 forks source link

Implement proper saving/loading of BertMesh to/from local #254

Open AndreiXYZ opened 1 year ago

AndreiXYZ commented 1 year ago

Currently we are not able to save/load to/from local disk. This is not an issue when we use the pretrained model from HF hub, but if we want to finetune the model additionally at some point, we will need this functionality.