search

agrimgupta92 / sgan

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

KeyError: 'g_state' #108

Closed Pradur241 closed 2 years ago

Pradur241 commented 2 years ago

After finished training the model, i met this problem when i tried to use the evaluate_model to test the checkpoint file.

Traceback (most recent call last): File "scripts/evaluate_model.py", line 117, in main(args) File "scripts/evaluate_model.py", line 106, in main generator = get_generator(checkpoint) File "scripts/evaluate_model.py", line 38, in get_generator generator.load_state_dict(checkpoint['g_state']) KeyError: 'g_state'

How can i solve the problem? Help please!

Pradur241 commented 2 years ago

Problem solved. Use checkpoint_with_model.pt