Missing key(s) in state_dict: "decoder.mlp.0.weight", "decoder.mlp.0.bias", "decoder.mlp.2.weight", "decoder.mlp.2.bias"

[foxfromworld]

Hi!
I set "pool_every_timestep" of the new TrajectoryGenerator (generator) in evaluate_model.py as "True" and an error occured:

Traceback (most recent call last):
File "scripts/evaluate_model.py", line 125, in
main(args)
File "scripts/evaluate_model.py", line 112, in main
generator = get_generator(checkpoint)
File "scripts/evaluate_model.py", line 41, in get_generator
generator.load_state_dict(checkpoint['g_state'])
File "....../lib/python3.5/site-packages/torch/nn/modules/module.py", line 721, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for TrajectoryGenerator:

Missing key(s) in state_dict: "decoder.mlp.0.weight", "decoder.mlp.0.bias", "decoder.mlp.2.weight", "decoder.mlp.2.bias".

When I changed the "pool_every_timestep" as "False", the error disappeared. Does this mean that I need to retrain the model for "pool_every_timestep=True" of the new TrajectoryGenerator for this evaluation setting?

Thank you!