search

fundamentalvision / BEVFormer

[ECCV 2022] This is the official implementation of BEVFormer, a camera-only framework for autonomous driving perception, e.g., 3D object detection and semantic map segmentation.
https://arxiv.org/abs/2203.17270
Apache License 2.0
3.4k stars 550 forks source link

请问在哪里设置训练batch_size #48

Closed lvxuanxuan123 closed 2 years ago

zhiqi-li commented 2 years ago

The samples_per_gpu is the batch_size on per GPU. When adjusting the batch size, please adjust the learning rate accordingly. https://github.com/zhiqi-li/BEVFormer/blob/846fb5bcf0ff746f4f50dfa436c7ccbf7e3122bd/projects/configs/bevformer_fp16/bevformer_tiny_fp16.py#L211

lvxuanxuan123 commented 2 years ago

The samples_per_gpu is the batch_size on per GPU. When adjusting the batch size, please adjust the learning rate accordingly.

https://github.com/zhiqi-li/BEVFormer/blob/846fb5bcf0ff746f4f50dfa436c7ccbf7e3122bd/projects/configs/bevformer_fp16/bevformer_tiny_fp16.py#L211

感谢李博士的工作！

lvxuanxuan123 commented 2 years ago

the issue has been solved