bf48c8a5741904f99b003ae1a8d718d283220f40,open_seq2seq/utils/funcs.py,,train,#Any#Any#Any#,14

Before Change

    hooks=hooks,
  ) as sess:
    if train_model.on_horovod:
      sess.run(train_model.get_data_layer().iterator.initializer)
    else:
      for i in range(train_model.num_gpus):
        sess.run(train_model.get_data_layer(i).iterator.initializer)
    step = 0

After Change

  if train_model.on_horovod:
    init_data_layer = train_model.get_data_layer().iterator.initializer
  else:
    init_data_layer = tf.group(
      [train_model.get_data_layer(i).iterator.initializer
       for i in range(train_model.num_gpus)]
    )

  scaffold = tf.train.Scaffold(
    local_init_op=tf.group(tf.local_variables_initializer(), init_data_layer)
  )

Instances