file “main_linear.py”， line103， set_model(opt) ，if torch.cuda.is_available() is false， why model will not load_state_dict(state_dict)

[wupeng1989]

` if torch.cuda.is_available():
.........
model = model.cuda()
classifier = classifier.cuda()
criterion = criterion.cuda()
cudnn.benchmark = True

    model.load_state_dict(state_dict)

`

[HobbitLong]

Thanks for pointing it out. you are right, that's bad engineering Habit. I was assuming people using this code will had GPU.

Added the else error message.