How can i use multi GPUs to train?

eg4000 / SKU110K_CVPR19

771 stars 182 forks source link

How can i use multi GPUs to train? #6

Closed MengNan-Li closed 5 years ago

MengNan-Li commented 5 years ago

When i use the train_iou.py, there are many warning.

like this:

...SKU110K_CVPR19/object_detector_retinanet/keras_retinanet/preprocessing/generatorIou.py:140: UserWarning: Image with id 5425 (shape (3264, 2448, 3)) contains the following invalid boxes: [array([2365., 2627., 2449., 2763., 0.])].

eg4000 commented 5 years ago

Hi,

I didn't try multiple GPU training. You can check this option in the RetinaNet original code.

The warnings are OK. It filters boxes which are close to the edges of the image.