ZoeM2016

大佬，我把batchsize已经设置为2，但是在多卡训练的时候还是出现了这个错误，该怎么办呢 raise ValueError('Expected more than 1 value per channel when training, got input size {}'.format(size)) ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256,...

sorry，i didn't put it very clearly.this code is in pvec.py line99 def normalize(self,smoother=0.0): s = self.sum() assert(s>=0) K = len(self.p) for i in range(K): self.p[i] = (self.p[i] + smoother)/(s +...