Warning with number of classes when evaluating the model on nuScenes

[shan18]

Hi, I downloaded the pre-trained model (open-loop planning on nuScenes) given in the README and tried to run the evaluation script on it:

bash scripts/eval_plan.sh <path to STP3_plan.ckpt> <path to nuScenes dataroot>

But when I run this script, I get the following warning:

.../python3.7/site-packages/pytorch_lightning/utilities/distributed.py:52: RuntimeWarning: You have set 2 number of classes which is different from predicted (1) and target (1) number of classes

When I checked the code, I see that the number of classes is loaded automatically from the config stored in the checkpoint. Can you please give me any pointers as to how I can resolve this warning?