search

DetectionTeamUCAS / FPN_Tensorflow

This is a tensorflow re-implementation of Feature Pyramid Networks for Object Detection.
https://github.com/DetectionTeamUCAS/FPN_Tensorflow
MIT License
348 stars 132 forks source link

about next_batch size #85

Closed chenhangcal closed 5 years ago

chenhangcal commented 5 years ago

https://github.com/DetectionTeamUCAS/FPN_Tensorflow/blob/9ea554665a127af72698ed7e3bedd5a3fd5db253/tools/multi_gpu_train.py#L118

Here batch_size=cfgs.BATCH_SIZE * num_gpu, but later only the first num_gpu images fed the network, if BATCH_SIZE >1, the rest of the images were wasted. So why not just set batch_size=num_gpu?

yangxue0827 commented 5 years ago

这个是为了方便以后多batch size和多gpu代码留的