HKUST-Aerial-Robotics / SIMPL

SIMPL: A Simple and Efficient Multi-agent Motion Prediction Baseline for Autonomous Driving
MIT License
215 stars 28 forks source link

vis #22

Open John122-art opened 5 months ago

John122-art commented 5 months ago

Hello,

Congratulations on achieving such excellent results. I have modified the code you provided by incorporating my ideas. However, during visualization and validation, I encountered an issue with "Missing key(s) in state_dict:". I presume the state dict was predefined by you, leading to this inconsistency. Therefore, I would like to ask where in the code the state dict is defined, what exactly is written into it, or how you view this issue. Thank you.

MasterIzumi commented 4 months ago

Hi, Do you add more layers to the network or modify the name of sub-modules? If so, the loading process cannot find the corresponding key of the weights, and the error will be raised.