search

ebanalyse / NERDA

Framework for fine-tuning pretrained transformers for Named-Entity Recognition (NER) tasks
MIT License
153 stars 35 forks source link

Save a trained model #16

Closed alfredcs closed 2 years ago

alfredcs commented 3 years ago

Is there a way to save a trained model?

yptheangel commented 3 years ago

14

smaakage85 commented 3 years ago

Yes, you can use 'torch' built-in functions, like proposed in #14:

image

I will build a method NERDA, that does this!

Best, Lars

smaakage85 commented 2 years ago

hi again @alfredcs

At long last I have implemented this (NERDA==1.0.0). See functions:

python model.save_network python model.load_network_from_file

Thank you so much for you feedback.