Isn’ that the same as model = model.cuda()? I mean I have tried to print out all the tensors device, they are all on cuda:0, but the GPU usage was pretty low and the speed didn’t improve really much as I used cpu