Hi, I have a model with a grid_sample 25 layer, I tried to train my model on multiple GPUs, but got the following error:
RuntimeError: grid_sampler(): expected input and grid to be on same device, but input is on cuda:1 and grid is on cuda:0

Is there anyway to use this layer on multiple GPUs? Thanks