Skip to content

Instantly share code, notes, and snippets.

@juliensimon

juliensimon/kerasmxnet-gpu.py

Last active May 30, 2018 17:32
Show Gist options
  • Save juliensimon/492e137a28ada394711b2a2218da478f to your computer and use it in GitHub Desktop.
Save juliensimon/492e137a28ada394711b2a2218da478f to your computer and use it in GitHub Desktop.
Download ZIP
kerasmxnet-gpu.py
Raw
kerasmxnet-gpu.py
from keras.utils import multi_gpu_model
model = Sequential()
model.add(...)
...
if gpu_count > 1:
model = multi_gpu_model(model, gpus=gpu_count)
model.compile(...)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment