Skip to content

Commit e469357

Browse files
committed
SentimentProcessor ought to be able to save the trainer now... will make it easier to access the whole model state elsewhere
1 parent a3513bf commit e469357

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

stanza/pipeline/sentiment_processor.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -55,6 +55,7 @@ def _set_up_model(self, config, pipeline, device):
5555
trainer = Trainer.load(filename=filename,
5656
args=args,
5757
foundation_cache=pipeline.foundation_cache)
58+
self._trainer = trainer
5859
self._model = trainer.model
5960
self._model_type = self._model.config.model_type
6061
# batch size counted as words

0 commit comments

Comments
 (0)