longcw / yolo2-pytorch

YOLOv2 in PyTorch
1.54k stars 421 forks source link

When I train pascval 2007,I meet out of memory #78

Closed gittigxuy closed 6 years ago

gittigxuy commented 6 years ago

default

I could not find where to change the value of batch_size,and then should I resize the voc data to (416*416)pixes?