Traffic-Light-Detection-Using-YOLOv3 icon indicating copy to clipboard operation
Traffic-Light-Detection-Using-YOLOv3 copied to clipboard

Published 20 hours ago • verified publisher icon sovit-123

Batch-size

Open ilpapds opened this issue 3 years ago • 1 comments

At the readme.md batch-size 2 is proposed. Isn't it really small for this dataset? Do you propose any other values for the batch size, e.g. 16, 32, 64 etc? Thank you in advance

ilpapds avatar Aug 01 '22 09:08 ilpapds

@ilpapds As I remember it it was to work around some GPU memory issue. If you can train with higher batch size, surely go for it.

sovit-123 avatar Aug 01 '22 15:08 sovit-123