Imageai: distributed training

Created on 17 Oct 2019  路  3Comments  路  Source: OlafenwaMoses/ImageAI

Does keras support distributed training? Can I use tensorflow's distributed training tools?

Most helpful comment

Distribution over multiple GPUs is already provided for custom object detection. For custom object there is no chance to use any other tool apart from setting different number of GPUs.

you can also edit the source code to apply them anyway

All 3 comments

You could use something like the 'BatBot' architecture to implement distributed training.
It captures images and associated training text for later batch submission.
Then, once new models are generated, it is easy to move them into place.

https://github.com/LeeHounshell/BatBot

Distribution over multiple GPUs is already provided for custom object detection. For custom object there is no chance to use any other tool apart from setting different number of GPUs.

you can also edit the source code to apply them anyway

I'm trying to distribution train imageai with horovod .

Was this page helpful?
0 / 5 - 0 ratings