Hi,
How can we reshape a tensor in state_dict? A call to model.state_dict() returns the state dictionary but any reshaping done does not affect the tensor in stat_dict object.
For example model.stat_dict()[‘weight’].view(-1,1) doesn’t work.
Hi,
How can we reshape a tensor in state_dict? A call to model.state_dict() returns the state dictionary but any reshaping done does not affect the tensor in stat_dict object.
For example model.stat_dict()[‘weight’].view(-1,1) doesn’t work.