有没有办法在几台机器上运行基于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,但这是一种更简单的方法吗?