示例命令执行时gpu不可用的问题

nju-websoft / muKG

μKG: A Library for Multi-source Knowledge Graph Embeddings and Applications, ISWC 2022

GNU General Public License v3.0

109 stars 11 forks source link

Closed AIR-hl closed 1 year ago

AIR-hl commented 1 year ago

X)IJQ NC%D(OZV}8S7378$X 3RY19G80Y{BRON 6 XZ)DPF 请问为什么按照示例命令执行，GPU状态可用，上面却输出的是 ea_models 的路径，device也变成 cpu 了

luoxindi commented 1 year ago

你好，在general_models.py文件的infer_model函数中，会导入进其他模型的进行初始化，使得某些模型会打印无关信息对您造成误解，稍后我会修改上传