Is model.cuda() or model.to(device) performed in place? Or do I need to reassign the model as such?
if torch.cuda.is_available():
    model = model.cuda()

or is this ok?
if torch.cuda.is_available():
    model.cuda()

Thank you in advance.