KeyError: 'all_seqeuence_tokens'

[croessert]

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]

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.