博主你好,请问我在实现那个在线医生的代码时:运行在ner_model的train.py文件时,出现以下错误:
Traceback (most recent call last):
File "train.py", line 281, in
LEARNING_RATE, tag_to_id, MODEL_SAVED_PATH, TRAIN_LOG_PATH, VALIDATE_LOG_PATH, TRAIN_HISTORY_IMAGE_PATH)
File "train.py", line 87, in train
loss = model.neg_log_likelihood(inputs, labels)
File "/usr/zhang_tao_code/online_doctor/ner_model/bilstm_crf.py", line 237, in neg_log_likelihood
feats = self._get_lstm_features(sentence)
File "/usr/zhang_tao_code/online_doctor/ner_model/bilstm_crf.py", line 102, in _get_lstm_features
lstm_out, self.hidden = self.lstm(embeds, self.hidden)
File "/root/anaconda3/envs/unit/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/anaconda3/envs/unit/lib/python3.7/site-packages/torch/nn/modules/rnn.py", line 772, in forward
self.check_forward_args(input, hx, batch_sizes)
File "/root/anaconda3/envs/unit/lib/python3.7/site-packages/torch/nn/modules/rnn.py", line 697, in check_forward_args
self.check_input(input, batch_sizes)
File "/root/anaconda3/envs/unit/lib/python3.7/site-packages/torch/nn/modules/rnn.py", line 212, in check_input
self.input_size, input.size(-1)))
RuntimeError: input.size(-1) must be equal to input_size. Expected 200, got 100
请问该怎么解决