edhenry / chexnet

Implementation and fullstack pipeline for CheXNet classifier
MIT License
13 stars 12 forks source link

Multi-GPU Support #1

Open edhenry opened 6 years ago

edhenry commented 6 years ago

Current model training only supports single GPU. Support for training utilizing multiple GPU's should be implemented.

Currently Keras has support for handling, natively, mutli-GPU training within the framework : https://keras.io/utils/#multi_gpu_model

Also adding the specific PR that was pushed into Keras for reference : https://github.com/keras-team/keras/pull/9226