search

agrimgupta92 / sgan

Code for "Social GAN: Socially Acceptable Trajectories with Generative Adversarial Networks", Gupta et al, CVPR 2018
MIT License
819 stars 260 forks source link

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

Open foxfromworld opened 5 years ago

foxfromworld commented 5 years ago

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!