search

dog-qiuqiu / MobileNet-Yolo

MobileNetV2-YoloV3-Nano: 0.5BFlops 3MB HUAWEI P40: 6ms/img, YoloFace-500k:0.1Bflops 420KB:fire::fire::fire:
Other
1.71k stars 277 forks source link

Question: batch size #10

Closed sealedtx closed 4 years ago

sealedtx commented 4 years ago

@dog-qiuqiu Hi! Thank you for your work. Please clarify why batch is so small (16) comparing to default yolov3 (64): https://github.com/dog-qiuqiu/MobileNetv2-YOLOV3/blob/2cda47a036e0378cb6c988326c3b53016eeb2b4f/MobileNetV2-YOLOv3-SPP/MobileNetV2-YOLOv3-SPP.cfg#L2

dog-qiuqiu commented 4 years ago

Because I have x8 GPUs, my batch size is X8. I suggest that the batch size should not be less than 128

sealedtx commented 4 years ago

@dog-qiuqiu Great, thanks for a quick answer!