Skip to content

Instantly share code, notes, and snippets.

@Damacustas
Damacustas / ModelMGPU.py
Created June 22, 2018 11:44
Multi GPU keras model
# Taken from https://github.com/keras-team/keras/issues/2436#issuecomment-354882296
# but saved into this gist to make it more easily findable.
from keras import Model
from keras.utils import multi_gpu_model
class ModelMGPU(Model):
def __init__(self, ser_model, gpus):
pmodel = multi_gpu_model(ser_model, gpus)