jwyang / fpn.pytorch

Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
MIT License
952 stars 221 forks source link

train with multi gpus can not work #52

Open mynameischaos opened 4 years ago

mynameischaos commented 4 years ago

Hi, I use multi gpus to train but not work.

Add option --cuda --mGPUs, I am sure that there are multi gpus to be available.

Can you help me ?