The pytorch gpu model is loaded into the CPU that do different results each time in test

[spmimi (shang)]

I think that my problems is like

Different results on cpu and gpu

I trained my model on GPU and saved the model with params = {k:v.cpu() for k, v in self.state_dict().items()} torch.save(params, open('model.pt', 'wb')) Then, when evaluating the model, I load it like this: model = m(...) params = torch.load('model.pt') model.load_state_dict(params) device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') model = model.to(device) However, I am getting much worse results on GPU, as on CPU. It seems like on CPU its performing correctly (I get sim…

can you see that? thanks