ConvE
ConvE copied to clipboard
Question about the batch-size
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