Hi,I wanna apply weights data to model.
weight=torch.load(weights_file)
print(weight)
self.model.load_state_dict(weight)
print(self.model.state_dict())

This torch.Tensor which name is ‘weight’ has  “device=‘cuda:0’”.
But “self.model.state_dict()” doesn’t have it.
Why?