OpenDriveLab / ST-P3

[ECCV 2022] ST-P3, an end-to-end vision-based autonomous driving framework via spatial-temporal feature learning.
Apache License 2.0
315 stars 38 forks source link

Warning with number of classes when evaluating the model on nuScenes #16

Open shan18 opened 1 year ago

shan18 commented 1 year ago

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?