deleted bottleneck in .cuda()

ZitongYu / CDCN

Central Difference Convolutional Networks (CVPR'20)

Other

553 stars 179 forks source link

deleted bottleneck in .cuda() #64

Open YaphetS7 opened 1 year ago

YaphetS7 commented 1 year ago

kernel_filter was computed two times: first for out, second for label. There was a bottleneck in first branch - the reason is .cuda() method. Now there are no bottlenecks in the code.