Hi,
I am getting the below error for the code which was just working, the network is the same, and now I get this error. Any possible solution? thanks.
Traceback (most recent call last):
  File "sample.py", line 70, in <module>
    decode(args)
  File "sample.py", line 33, in decode
    decoder.load_state_dict(torch.load(args.decoder_path))
  File "/home/rj/anaconda2/envs/dlp/lib/python3.6/site-packages/torch/nn/modules/module.py", line 769, in load_state_dict
    self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for DecoderRNN:
	Missing key(s) in state_dict: "embed.weight", "lstm.weight_ih_l0", "lstm.weight_hh_l0", "lstm.bias_ih_l0", "lstm.bias_hh_l0", "lstm.weight_ih_l1", "lstm.weight_hh_l1", "lstm.bias_ih_l1", "lstm.bias_hh_l1". 
	Unexpected key(s) in state_dict: "bn.weight", "bn.bias", "bn.running_mean", "bn.running_var", "bn.num_batches_tracked". 
	size mismatch for linear.weight: copying a param with shape torch.Size([256, 2048]) from checkpoint, the shape in current model is torch.Size([36881, 512]).
	size mismatch for linear.bias: copying a param with shape torch.Size([256]) from checkpoint, the shape in current model is torch.Size([36881]).