Skip to content

Commit

Permalink
SentimentProcessor ought to be able to keep 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 a17029c
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 a17029c

Please sign in to comment.