测试的一个小 bug
ludics opened this issue · comments
Lu Di commented
测试的时候,如果测试集的数量和 batch size 的设置使得最后一个 batch 中只有一张图片时,会出现 invalid index to scalar variable 这样的 IndexError.
这是因为 test.py 第 154 行:
node_list = node_list.long().squeeze().numpy()
bs = feat.size(0)
产生的 node_list 在此时会退化成一维数组,导致索引错误.
可以再加一句检查:
node_list = node_list.long().squeeze().numpy()
bs = feat.size(0)
if bs == 1:
node_list = np.array([node_list])