metadriverse / cat Goto Github PK

Dear Authors,

thanks in advance!
As mentioned in the title, I wonder how much GPU RAM does it need to train? I have a GPU with 11GB RAM, but it always stop with CUDA out of memory?

I used raw_scenes_500 provided, it works fine. But when generating my own data from waymo scenarios validation/testing_interactive, there is always KeyError: 'DRIVEWAY' or 'polyline'. Do you have any idea

Hi, the main readme mentions the modified Metadrive. I was wondering if that is specified somewhere? Thank you!

Hello,

Thank you for the valuable work! Could you please provide the code for adversarial scenario generation for the IDM/Pretrained planner (Table 1 in the paper), or describe how it is implemented?

Thanks a lot for the excellent work and Open-sourced code repo.
When running the training process, i got following error. Do you have any idea?
Traceback (most recent call last):
File "/home/xliu/ws_cat/cat/cat_RLtrain.py", line 233, in
adv_generator.generate()
File "/home/xliu/ws_cat/cat/advgen/adv_generator.py", line 423, in generate
trajs_OV = pred_trajectory[1]
IndexError: index 1 is out of bounds for axis 0 with size 1