diff options
Diffstat (limited to 'stanza/models/ner/model.py')
| -rw-r--r-- | stanza/models/ner/model.py | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/stanza/models/ner/model.py b/stanza/models/ner/model.py index efad8d51..dbd93b35 100644 --- a/stanza/models/ner/model.py +++ b/stanza/models/ner/model.py @@ -1,3 +1,4 @@ +import os import numpy as np import torch import torch.nn as nn @@ -35,6 +36,10 @@ class NERTagger(nn.Module): if self.args['char'] and self.args['char_emb_dim'] > 0: if self.args['charlm']: + if args['charlm_forward_file'] is None or not os.path.exists(args['charlm_forward_file']): + raise FileNotFoundError('Could not find forward character model: {} Please specify with --charlm_forward_file'.format(args['charlm_forward_file'])) + if args['charlm_backward_file'] is None or not os.path.exists(args['charlm_backward_file']): + raise FileNotFoundError('Could not find backward character model: {} Please specify with --charlm_backward_file'.format(args['charlm_backward_file'])) add_unsaved_module('charmodel_forward', CharacterLanguageModel.load(args['charlm_forward_file'], finetune=False)) add_unsaved_module('charmodel_backward', CharacterLanguageModel.load(args['charlm_backward_file'], finetune=False)) input_size += self.charmodel_forward.hidden_dim() + self.charmodel_backward.hidden_dim() |