RetroExplainer icon indicating copy to clipboard operation
RetroExplainer copied to clipboard

练50K数据集生成的leaving_group的维度与测试时的维度不匹配

Open l727248517 opened this issue 2 years ago • 2 comments

现有代码执行训练50K数据集生成的leaving_group的维度与测试时的维度不匹配,我将datamodule中50k的max_node/max_lg_na/get_num_size修改为50、23、4之后训练生成的leaving_group在运行测试脚本没有报错,但是测试结果的精度比预期低很多,是否我这么改是错误的。 Uploading image.png…

l727248517 avatar Dec 21 '23 11:12 l727248517

你好,请问你是直接加载的预训练模型?未能复现效果的原因可能是leaving_group索引与预训练模型的leaving_group的索引发生了错位,README文本里有提供已经处理好的leaving_group,建议使用直接处理好的内容。

wangyu-sd avatar Aug 08 '24 11:08 wangyu-sd

邮件已收到,谢谢!~                            _lwy。

l727248517 avatar Aug 08 '24 11:08 l727248517