Tensorflow aggregation_method for optimizers

Viewed 1952

I could not find documentation regarding the aggregation method in tensorflow optimizer

I have the following line of code

train_op = optimizer.minimize(loss, global_step=batch, aggregation_method = tf.AggregationMethod.EXPERIMENTAL_TREE)

However, this property can be changed to be

tf.AggregationMethod.EXPERIMENTAL_ACCUMULATE_N

Does anyone know what does it do? (I just know that when I used the default with an LSTM it did not have enough memory to run)

1 Answers
Related