I don’t see .cuda() or (‘cuda’) operations anywhere in the source code… Do you? In trainer.py i do see this though:
def get_model_device(net):
device = torch.device(“cuda:0”)
#for param in net.parameters():
#    device = param.device
#    break
return device