I want to do this:
model=Net()
model.cuda() ##Convert everything to cuda()

I dont want to use .cuda() as later on i will be migrating the code to a CPU only device where i would need to test on CPU.
EDIT: I will train on GPU and test on CPU, so the weights will need to be .cpu()