Могу ли я запустить 2 сеанса параллельно, используя Tensorflow V1? Правильно ли писать программу следующим образом?

Я хотел бы оптимизировать 2 оптимизатора/функции, используя тензорный поток v1 с использованием обучающего набора, поэтому хотел бы запустить два оптимизатора параллельно.

      import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()

# ... 

loss_i1 = - tf.reduce_mean(s1+s2+s3)
optimizer_i1 = tf.train.AdamOptimizer(learning_rate).minimize(loss_i1)
loss_i2 = - tf.reduce_mean(s1*s2*s3)
optimizer_i2 = tf.train.AdamOptimizer(learning_rate).minimize(loss_i2)

for epoch in range(epochs):
    for batch_index in range(n_batches):
        sys.stdout.flush()
        _y, _s1, _s2, _s3 = fetch_batch(epoch, batch_index, batch_size, y_, s1_, s2_, s3_)
        sess.run(optimizer_i1, feed_dict={y: _y, s1: _s1, s2: _s2, s3: _s3})   
        sess.run(optimizer_i2, feed_dict={y: _y, s1: _s1, s2: _s2, s3: _s3})   
    loss_train_i1 = loss_i1.eval(feed_dict={y: _y, s1: _s1, s2: _s2, s3: _s3})
    loss_train_i2 = loss_i2.eval(feed_dict={y: _y, s1: _s1, s2: _s2, s3: _s3})

0 ответов

Другие вопросы по тегам