我正在尝试保存 LayoutLM 模型的预测。
论文 - https://arxiv.org/abs/1912.13318
笔记本 - https://www.kaggle.com/code/iamarjunchandra/layoutlm-document-sequence-labeling-model/notebook
The problem arises when I run the below code for making prediction
! python unilm/layoutlm/examples/seq_labeling/run_seq_labeling.py
--do_predict
--data_dir data
--model_type layoutlm
--model_name_or_path output
--output_dir output
--labels data/labels.txt \
--fp16
我收到以下错误。
Traceback (most recent call last):
File "unilm\\layoutlm\\examples\\seq_labeling\\run_seq_labeling.py", line 53, in <module>
from layoutlm import FunsdDataset, LayoutlmConfig, LayoutlmForTokenClassification
ImportError: cannot import name 'LayoutlmConfig' from 'layoutlm' (c:\Users\jyoti\anaconda3\lib\site-packages\layoutlm\__init__.py)
你如何解决这个问题?