当batchsize设为4或8时报错

[geolvr]

train_sample.py中的np.random.choice(np.arange(0, num_parts), 1, p=part_weights) 处报错 “probabilities contain NaN”. 使用batchsize=1, 12, 16, 32均不报错。请问是为什么呢？该算法对batchsize有特殊限制？