Loading model with low accuracy

[posenhuang]

Hi!
I found that when I load the model back using autoencoder.load_state_dict(torch.load(ae_path)) etc., it leads to low accuracy (around 0.3) even though the model achieves around 0.8 accuracy in evaluate_autoencoder during training.
I have to use torch.load(ae_path) and torch.save(autoencoder, f) to get away this problem.
Probably it is a pytorch bug, as discussed here: https://discuss.pytorch.org/t/saving-and-loading-a-model-in-pytorch/2610/6

[thvasilo]

@posenhuang Did you do model.eval() after loading the model to disable dropout?

[posenhuang]

Yes, I did, but it it still the same. I'll double check again.

[lan2720]

@posenhuang I met the similar problem. Did you solve it?