search

XinJCheng / CSPN

Convolutional Spatial Propagation Network
496 stars 92 forks source link

cspn求和的卷积核应该在forward的时候创建 #22

Closed yellowYuga closed 5 years ago

yellowYuga commented 5 years ago

self.weight = torch.ones(1, 8, 1, 1, 1).cuda() cspn中的求和卷积如果在init过程构建的话，batch运行的时候，pytorch会提示不在一张卡上。 所以该weight应该在forward的过程中创建

XinJCheng commented 5 years ago

yes, this only works for one GPU training, thanks for your kindly reminder, I have updated it.