KeyError: 'all_seqeuence_tokens'

HorizonRobotics / GUMP

Generative model for Unified Motion Planning tasks

Apache License 2.0

121 stars 8 forks source link

KeyError: 'all_seqeuence_tokens' #3

Open croessert opened 3 months ago

croessert commented 3 months ago

Hi, thanks for your work!

I am getting the following error in scripts/training/train_wod_model.sh just after the validation step:

File "GUMP/nuplan_extent/planning/training/callbacks/visualization_vectorized_reconstruction_callback.py", line 235, in _log_batch all_seqeuence_tokens = predictions['all_seqeuence_tokens'] KeyError: 'all_seqeuence_tokens'

Yihanhu commented 3 months ago

Hi @croessert , Sorry for the confusion caused. Please try setting this argument to something other than 'planning' or 'sim_agents' during training. This part might indeed cause some bugs during the validation stage, and I will fix it later.