def testAddCounter(self): a = tf.placeholder(tf.int64, shape=[]) hooks.add_counter("sum_a", a) sum_a = tf.get_collection(hooks._DEFAULT_COUNTERS_COLLECTION) self.assertIsNotNone(sum_a) with self.test_session() as sess: sess.run(tf.global_variables_initializer()) self.assertEqual([6], sess.run(sum_a, feed_dict={a: 6})) self.assertEqual([10], sess.run(sum_a, feed_dict={a: 4}))
def _register_word_counters(self, features, labels): """Creates word counters for sequences (if any) of :obj:`features` and :obj:`labels`. """ features_length = self._get_features_length(features) labels_length = self._get_labels_length(labels) with tf.variable_scope("words_per_sec"): if features_length is not None: add_counter("features", tf.reduce_sum(features_length)) if labels_length is not None: add_counter("labels", tf.reduce_sum(labels_length))