问题描述
我试图利用https://github.com/microsoft/unilm/tree/master/layoutlm进行文档分类,但一直得到“ OSError:无法从pytorch检查点文件中加载权重。”
有人可以帮我运行和使用layoutLM吗?
配置/版本:
- Windows 10
- Python-3.6.5
- huggingface-transformers-3.1.0
- 火炬-1.5.0
- tensorflow-2.3.1
运行代码的命令:
python run_classification.py --data_dir C:\Users\Downloads\unilm-master\unilm-master\layoutlm\examples\classification\data --model_type layoutlm --output_dir C:\Users\Downloads\unilm-master\unilm-master\layoutlm\examples\classification\data --do_eval --model_name_or_path
我相信命令--model_name_or_path
存在一些问题,我尝试了上述方法,并尝试下载pytorch_model.bin文件进行layoutlm并将其指定为--model_name_or_path
的参数,但是没有救命。
C:\ Users \ Downloads \ unilm-master \ unilm-master \ layoutlm \ examples \ classification \ model \ pytorch_model.bin。
我也怀疑是否是因为Transformer支持与layoutlm支持(与Transformers版本3.1.0或2.0.0有关)之间的差异?
有人可以帮助我快速掌握layoutLM吗?
感谢您的帮助。
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com (将#修改为@)