Ошибка при обучении NER с TensorFlow

1

Добрый день!
Пытаюсь обучить модель ner_rus на собственных данных с собственными типами сущностей.
Однако, получаю
2021-12-15 15:46:52.739 ERROR in ‘deeppavlov.core.common.params’[‘params’] at line 112: Exception in <class ‘deeppavlov.models.ner.network.NerNetwork’>
Traceback (most recent call last):
File “/home/s/.local/lib/python3.6/site-packages/tensorflow_core/python/client/session.py”, line 1365, in _do_call
return fn(*args)
File “/home/s/.local/lib/python3.6/site-packages/tensorflow_core/python/client/session.py”, line 1350, in _run_fn
target_list, run_metadata)
File “/home/s/.local/lib/python3.6/site-packages/tensorflow_core/python/client/session.py”, line 1444, in _call_tf_sessionrun
run_metadata)
tensorflow.python.framework.errors_impl.InvalidArgumentError: Assign requires shapes of both tensors to match. lhs shape= [256,3] rhs shape= [256,7]
[[{{node save/Assign_14}}]]

В чем может быть причина ? Где искать ошибку ?
Спасибо.

2

Добрый день,
эта же ошибка обсуждалась в issues на github: Assign requires shapes of both tensors to match while train_model ner_ontonotes_bert_mult · Issue #839 · deepmipt/DeepPavlov · GitHub

Чтобы обучить свою модель на своих данных надо поменять все пути до модели и словаря с лейблами (параметры save/load path).