Assuming my model is on a gpu already, is there a way to get a state_dict of a model with cpu tensors without moving the model first to cpu and then back to gpu again?
Something like:
state_dict =  model.state_dict()
state_dict = state_dict.cpu()