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?
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:
But when I run this script, I get the following warning:
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?