def begin(self): self._counters = tf.get_collection("counters") if not self._counters: return if self._summary_writer is None and self._output_dir: self._summary_writer = SummaryWriterCache.get(self._output_dir) self._last_count = [None for _ in self._counters] self._global_step = tf.train.get_global_step() if self._global_step is None: raise RuntimeError("Global step should be created to use WordCounterHook.")
def begin(self): self._counters = tf.get_collection("counters") if not self._counters: return if self._summary_writer is None and self._output_dir: self._summary_writer = SummaryWriterCache.get(self._output_dir) self._last_count = [None for _ in self._counters] self._global_step = tf.train.get_global_step() if self._global_step is None: raise RuntimeError("Global step should be created to use WordCounterHook.")
def __init__(self, labels_file=None, output_dir=None): self._labels_file = labels_file self._summary_writer = None if output_dir is not None: self._summary_writer = SummaryWriterCache.get(output_dir)