def setUp(self): """Mock out logging calls to verify if correct info is being monitored.""" self._logger = mock_lib.MockBenchmarkLogger() self.graph = tf.Graph() with self.graph.as_default(): tf.train.create_global_step() self.train_op = tf.assign_add(tf.train.get_global_step(), 1) self.global_step = tf.train.get_global_step()
def setUp(self): super(LoggingMetricHookTest, self).setUp() self._log_dir = tempfile.mkdtemp(dir=self.get_temp_dir()) self._logger = mock_lib.MockBenchmarkLogger()