Question about the batch-size

[zhanking321]

Hello, I want to ask about the batch-size. When I change the batch-size to 512, it has some problems:
Traceback (most recent call last):
File "/root/anaconda3/lib/python3.5/threading.py", line 914, in _bootstrap_inner
self.run()
File "/home/disk_sdb/app/ConvE-master/src/spodernet/spodernet/preprocessing/batching.py", line 175, in run
shard_idx = self.rdm.choice(len(list(self.shard2batchidx.keys())), 1, p=self.shard_fractions)[0]
File "mtrand.pyx", line 917, in numpy.random.mtrand.RandomState.choice
ValueError: 'a' and 'p' must have same size