garlicbulb-puzhuo / ggo

1 stars 0 forks source link

Figure out distributed tensorflow #58

Open tomowind opened 7 years ago

tomowind commented 7 years ago

TODO (read later):

tomowind commented 7 years ago

@oakcreek Googled with keras with distributed tensorflow, and found this: https://github.com/fchollet/keras/issues/1833

It seems keras doesn't use distributed tensorflow, and elephas seems the best option? To use distributed tensorflow, we need to write code in tensorflow.

tomowind commented 7 years ago

@oakcreek I was wrong. See Keras as a simplified interface to TensorFlow: tutorial. It's very simply done with 4 lines of code.

server = tf.train.Server.create_local_server()
sess = tf.Session(server.target)

from keras import backend as K
K.set_session(sess)