从 train.py 生成的保存文件中进行模型推理

我已经准备了一个自定义标注数据集来微调 DONUT, 专门用于发票数据提取。 train.py 文件运行正常, 在验证数据集上提供了良好的结果, 该数据集是训练脚本的一部分。 但是, 我无法从保存的文件中恢复经过微调的模型。 我遇到的错误包括 'SwinTransformer' 对象没有属性 'pos_drop', '您尝试加载的检查点具有模型类型 `donut`, 但 Transformers 不能识别此架构。 这可能是由于检查点存在问题, 也可能是因为您的 Transformers 版本过时。' 以及一些关键不匹配问题。 我还尝试克隆整个仓库, 并运行 train 和 test 命令, 如他们所提供的那样。