I don’t really understand the question.
If you want to use the GPU with your model you have to allocate both, model and input/gt in the gpu using .cuda()
If you don’t do it, it will display an error like tensor and model are not in the same device