关于数据集组织的问题
zhilanyue opened this issue · 2 comments
你好!
在复现你的代码的时候,执行sfrs训练(./scripts/train_sfrs_dist.sh)遇到了这样的问题:
问题出在(
OpenIBL/ibl/utils/data/sampler.py
Line 85 in 5ab80d6
在采样器中输出发现训练集的pos_list\neg_list是一样的:
OpenIBL/ibl/utils/data/sampler.py
Line 32 in 5ab80d6
OpenIBL/ibl/utils/data/sampler.py
Line 33 in 5ab80d6
修改过dataset.py中近邻搜索方法搜索半径,问题依然存在
OpenIBL/ibl/utils/data/dataset.py
Line 28 in 5ab80d6
neg_list
存的是25m以内的图,然后在sampler里会针对每个anchor的neg_list找到25m以外的negative pair,参见https://github.com/yxgeee/OpenIBL/blob/master/ibl/utils/data/sampler.py#L79
neg_list
存的是25m以内的图,然后在sampler里会针对每个anchor的neg_list找到25m以外的negative pair,参见https://github.com/yxgeee/OpenIBL/blob/master/ibl/utils/data/sampler.py#L79
了解,感谢!