使用Keras和fit_generator

时间:2017-06-22 22:48:01

标签: tensorflow keras distributed-computing

有没有办法在几台机器上运行基于model.fit_generator()的Keras模型?

我使用

定义了我的模型
model = Sequential()
model.add(...)

所以我没有对我的tf.Variables的低级访问权限,也无法使用

with tf.device("/job:ps/task:1"):
  weights_2 = tf.Variable(...)
  biases_2 = tf.Variable(...)

with tf.device(tf.train.replica_device_setter(
    worker_device="/job:worker/task:%d" % FLAGS.task_index,
    cluster=cluster)):

Google's distributed tensor flow tutorial中所述。

我找到了一种方法using Keras + Spark,但这是一种更简单的方法吗?

0 个答案:

没有答案