Why samples_per_gpu is set to 1?

OpenDriveLab / UniAD

[CVPR'23 Best Paper Award] Planning-oriented Autonomous Driving

Apache License 2.0

3.11k stars 335 forks source link

Why samples_per_gpu is set to 1? #150

Closed buaazeus closed 6 months ago

buaazeus commented 7 months ago

Can it be samples_per_gpu = 2? Thank you.

YTEP-ZHI commented 6 months ago

It means batch_size is 1 in training. You cannot set it to 2 (or higher) in the current version due to the complex data/tensor reshaping operations in each module. We will clean up the code to support larger batch sizes in the future.