search

pengfei-luo / MIMIC

[KDD 2023] Multi-Grained Multimodal Interaction Network for Entity Linking
18 stars 3 forks source link

模型能够训练，但val评测的时候进程自动被kill了，是内存不够？ #3

Open xgl0626 opened 1 year ago

xgl0626 commented 1 year ago

机器是v100*32g，作者你训练和val评测的时候内存是多大呢

pengfei-luo commented 1 year ago

你好，我运行代码的服务器内存为786GB，实际上内存占用(REZ)大概在10G以下。

xgl0626 commented 12 months ago

你好，我运行代码的服务器内存为786GB，实际上内存占用(REZ)大概在10G以下。

谢谢作者回复，已经跑通， 经过复现发现，训练内存占用不大，训练之后的eval过程中需要较大的内存。然后还有一个疑问，论文里面提到GHMFC模型是重新复现训练，为啥您们团队复现出来的模型精度比原论文中的精度提升很大，是因为数据清洗的原因吗

pengfei-luo commented 11 months ago

你好，GHMFC和原始论文结果有区别的原因可能为实验设置不相同。GHMFC中的KG设置为“The KG used in this paper is a subset extracted from Wikidata, and has more than 170K triples and about 80K entities, which contains all entities in datasets.”。但并没有提供相关KG数据的下载（此处提供下载的KG是wikidata最近的dumps），此处采用的KG设置和原文不同。希望可以解答你的疑惑。