chuanenlin / drone-net

https://towardsdatascience.com/tutorial-build-an-object-detection-system-using-yolo-9a930513643a
154 stars 66 forks source link

batch size #10

Open Ginbor opened 5 years ago

Ginbor commented 5 years ago

how do you think, what is the minimum train batch size?

chuanenlin commented 4 years ago

Great post regarding tradeoffs of different batch size. Also, if your GPU runs out of memory, you can try reducing the batch size.