search

zjukg / MyGO

[Paper][Preprint 2024] MyGO: Discrete Modality Information as Fine-Grained Tokens for Multi-modal Knowledge Graph Completion
https://arxiv.org/abs/2404.09468
210 stars 4 forks source link

请问跑MKG-W数据集图像和文本的token embedding在哪找呢 #21

Open teatimekon opened 2 weeks ago

Zhang-Each commented 2 weeks ago

你好，就在tokens目录下面，MKG-W开头的两个文件里。

teatimekon commented 2 weeks ago

你好，就在tokens目录下面，MKG-W开头的两个文件里。

谢谢你，那视觉和文本的嵌入也是用 visual.pth 和 save_token_embedding 生成的texual.pth 吗

Zhang-Each commented 2 weeks ago

是的，因为在不同的数据集上是统一用的这两个.pth文件里的token作为code book，他们和具体的知识图谱无关，是对应预训练模型BEiT和BERT的产物。