Tensorflow reset or clear collection

Question

In tensorflow, I find the API tf.add_to_collcetion to add some value to collection like code bellow.

def accuracy_rate(logits, labels):
    correct = tf.nn.in_top_k(logits, labels, 1)
    # Return the accuracy of true entries.
    accuracy = tf.reduce_mean(tf.cast(correct, tf.float32))
    return accuracy 
with tf.Session() as sess:
    logits, labels = ...
    accuracy = accuracy_rate(logits, labels)
    tf.add_to_collection('total_accuracy', sess.run(accuracy))

What I can't find in the API is that, how can I clear all values that I've already stored in one collection?

Answer 1

You can use tf.get_collection_ref to get a mutable reference to the collection which you can clear (it's just a python list).