search

csuhan / ReDet

Official code of the paper "ReDet: A Rotation-Equivariant Detector for Aerial Object Detection" (CVPR 2021)
https://redet.csuhan.com
Apache License 2.0
391 stars 78 forks source link

bach_size设置问题 #80

Open 1994su opened 3 years ago

1994su commented 3 years ago

请问一下咱们模型训练的batchsize在哪里修改？没有看到相关参数，目前用的是8卡 gpu，设置多少batchsize合适？

csuhan commented 3 years ago

Batch_size: https://github.com/csuhan/ReDet/blob/0b9addf3c2734659fd6ffc7824f2e659fde4419c/configs/ReDet/ReDet_re50_refpn_1x_dota1.py#L142

Copied from https://github.com/csuhan/ReDet/blob/master/README.md

All our models are trained on 4GPUs with a learning rate 0.01. If you train your model with more/fewer GPUs, remember to change the learning rate, e.g., 0.02lr=0.0025lr8GPU, 0.0025lr=0.0025lr1GPU.