Closed yellowYuga closed 5 years ago
self.weight = torch.ones(1, 8, 1, 1, 1).cuda() cspn中的求和卷积如果在init过程构建的话,batch运行的时候,pytorch会提示不在一张卡上。 所以该weight应该在forward的过程中创建
yes, this only works for one GPU training, thanks for your kindly reminder, I have updated it.
self.weight = torch.ones(1, 8, 1, 1, 1).cuda() cspn中的求和卷积如果在init过程构建的话,batch运行的时候,pytorch会提示不在一张卡上。 所以该weight应该在forward的过程中创建