Skip to content

Commit

Permalink
SentimentProcessor ought to be able to save the trainer now... will m…
Browse files Browse the repository at this point in the history
…ake it easier to access the whole model state elsewhere
  • Loading branch information
AngledLuffa committed Oct 27, 2024
1 parent a3513bf commit e469357
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions stanza/pipeline/sentiment_processor.py
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@ def _set_up_model(self, config, pipeline, device):
trainer = Trainer.load(filename=filename,
args=args,
foundation_cache=pipeline.foundation_cache)
self._trainer = trainer
self._model = trainer.model
self._model_type = self._model.config.model_type
# batch size counted as words
Expand Down

0 comments on commit e469357

Please sign in to comment.