search

zzh8829 / yolov3-tf2

YoloV3 Implemented in Tensorflow 2.0
MIT License
2.51k stars 905 forks source link

Add multi-gpu functionality #366

Closed jbutle55 closed 3 years ago

jbutle55 commented 3 years ago

These changes allow for training to be completed using more than 1 GPU. Using multiple GPUs is optional, defined by a new boolean argument. This was to address a few Issues raised discussing/requesting multiple GPU.

A couple chunks of code got moved around to implement this without having to copy code in multiple spaces. I also added tracking training time.

I successfully test ran these changes on Ubuntu 18.04 LTS (Bionic Beaver) using 2 GPU.

If you like the idea and want anything changed in the pull request feel free to suggest.

zzh8829 commented 3 years ago

Hello thanks so much for the contribution just one little question for you there