gather torch_torch.nn.DataParallel中数据Gather的问题:维度不匹配
Pytorch中的多GPU非常好用,一句话就能搞定:self.model = torch.nn.DataParallel(self.model)。然而这两天我做零样本学习中遇到了一个问题始终无法解决,就是说单GPU可以跑,一旦使用多GPU,就会出现:RuntimeError: Gather got an input of invalid size: got [24, 10, 448,448], b...