search

zjukg / NeuralKG

[Tool] For Knowledge Graph Representation Learning
http://neuralkg.zjukg.org/
Apache License 2.0
348 stars 63 forks source link

在cpu版linux上操作和win7版GPU上操作遇到的问题 #5

Closed sokolo05 closed 2 years ago

sokolo05 commented 2 years ago

cpu版上Linux上执行 sh scripts/FreeBase/NNE_FB.sh 报错如下：

Using backend: pytorch Traceback (most recent call last): File "main.py", line 8, in from neuralkg.utils import setup_parser File "/home/laicx/anaconda3/envs/neuralkg/lib/python3.8/site-packages/neuralkg/init.py", line 3, in from .lit_model import * File "/home/laicx/anaconda3/envs/neuralkg/lib/python3.8/site-packages/neuralkg/lit_model/init.py", line 1, in from .BaseLitModel import BaseLitModel File "/home/laicx/anaconda3/envs/neuralkg/lib/python3.8/site-packages/neuralkg/lit_model/BaseLitModel.py", line 6, in from neuralkg import loss File "/home/laicx/anaconda3/envs/neuralkg/lib/python3.8/site-packages/neuralkg/loss/init.py", line 8, in from .IterE_Loss import IterE_Loss ModuleNotFoundError: No module named 'neuralkg.loss.IterE_Loss'

在win7版GPU设备上使用pycharm执行 sh NNE_FB.sh 报错如下：

D:\ProgramData\Anaconda3\envs\neuralkg38\python.exe: can't open file 'main.py': [Errno 2] No such file or directory

sokolo05 commented 2 years ago

问题已经解决了，是自己对环境太不熟悉了

Modberge commented 2 years ago

你好，IterE_Loss目前已经在NeuralKG中弃用，因此如果遇到ModuleNotFoundError问题，可以选择删除from .IterE_Loss import IterE_Loss命令或更新NeuralKG。

当前NeuralKG默认使用GPU加速训练，采用CPU会极大延长训练时间。如果您当前需要使用CPU进行NeuralKG训练，只需将main.py的81行和91行命令改为gpus=0就可以简单实现，但请注意这种方法在一些模型中没有经过测试，因此我们还是推荐采用GPU加速模型训练的方法。