运行bash ./scripts/kgc/knnkge.sh时报错AttributeError: 'KNNKGEPretrainLitModel' object has no attribute 'model'

[Roger-Ni]

我想训练knn-kg模型，出现如上图所示错误。

[CheaSim]

请问一下transformers和pytorch_lightning的版本，以及模型文件是否成功下载。

[Roger-Ni]

transformers==4.15.0 pytorch_lighting==1.6.0 .模型文件的话，我看knnkge.sh文件中写的是model_name_or_path="bert-base-uncased"，就认为应该是下载bert的模型，我就下载到了如下图所示的目录下。

然后bert-base-uncased文件夹下的文件应该也是齐全的（如下图所示）

[CheaSim]

请把transformers 版本和requirements.txt 保持一致。

transformers==4.6.0

[zxlzr]

请问您的问题是否已解决

[Roger-Ni]

我把transformers和pytorch_lighting都配置成了rquirements中的版本，但是运行时，还是会出现如下错误。

[CheaSim]

这个貌似是transformers版本的问题，麻烦您使用python3.8 并且transformers更新到4.6.0.
请先执行

pip install -r requirements.txt

再运行我们的脚本~

[zxlzr]

请问您的问题是否已解决？