search

OpenDriveLab / UniAD

[CVPR 2023 Best Paper Award] Planning-oriented Autonomous Driving
Apache License 2.0
3.56k stars 405 forks source link

batch size set #184

Open ustcjasonliu opened 6 months ago

ustcjasonliu commented 6 months ago

It seems that multi batches is not supported now. I set samples_per_gpu to 2, but it went wrong. Did I set the wrong parameter?

generalchan825 commented 2 months ago

hi I faced the same issue. samples_per_gpu = 1, workers_per_gpu = 16 worked, but when changing samples_per_gpu = 12, it didn't work. From #150 we have the answer.

YTEP-ZHI commented 2 months ago

Hi @ustcjasonliu @generalchan825, unfortunately, it's not supported to set bs larger than 1 in the current implementation.

wlll123456 commented 3 weeks ago

Is this problem solved? Is it possible to increase the batch size?

YTEP-ZHI commented 3 weeks ago

Is this problem solved? Is it possible to increase the batch size?

Hi @wlll123456, sorry to inform you that larger batch size is still not supported due to our schedule. We sincerely welcome the contribution of the community to help on that.