Issue with torch.save/load

After training the UNet the weights are saved with `torch.save(model.state_dict(), 'file.pt')` and later the model `loaded_model.load_state_dict.load('file.pt', weights_only=True)` is created but `model` and `loaded_model` have different outputs. This problem did not arise with other models. Any idea what could be the cause?