search

Thinklab-SJTU / Bench2DriveZoo

BEVFormer, UniAD, VAD in CARLA under Closed-Loop Evaluation
44 stars 3 forks source link

VAD training time #5

Closed itsbman closed 1 week ago

itsbman commented 1 week ago

Hi, thank you for sharing this work!

I want to know how long does it typically take to train VAD for 60 epochs on the base dataset?

I'm training with 8 A6000 gpus and the eta is around 32 days

jiaxiaosong1002 commented 1 week ago

@itsbman As mentioned in paper, since the frequency of the dataset is 10Hz instead of 2Hz in nuScenes, there is lots of very similar samples. Thus, we only train 1/10 epochs - 6 epochs. The loss is normal.

jiaxiaosong1002 commented 1 week ago

@itsbman Hi, you may modify the lr, epoch_number, and dropout as in the updated config file.

itsbman commented 1 week ago

That makes sense. Thanks