Hi @ptrblck
Thanks. Actually the feature map is in size [64, 512, 49, 49]. And by applying those values it doesn’t work/
conv= nn.Conv2d(512,512,kernel_size=25)
SO_fmaps_extend = conv(fmaps)
*** RuntimeError: Input type (CUDAFloatTensor) and weight type (CPUFloatTensor) should be the same