def test_eager(self): with tf.compat.v2.summary.create_file_writer(self.logdir).as_default(): result = hp.hparams( self.hparams, trial_id=self.trial_id, start_time_secs=self.start_time_secs, ) self.assertTrue(result) self._check_logdir(self.logdir)
def test_graph_mode(self): with tf.compat.v1.Graph().as_default(), tf.compat.v1.Session( ) as sess, tf.compat.v2.summary.create_file_writer( self.logdir).as_default() as w: sess.run(w.init()) summ = hp.hparams(self.hparams, start_time_secs=self.start_time_secs) self.assertTrue(sess.run(summ)) sess.run(w.flush()) self._check_logdir(self.logdir)
def test_eager_no_default_writer(self): result = hp.hparams(self.hparams, start_time_secs=self.start_time_secs) self.assertFalse(result) # no default writer
def on_train_begin(self, logs=None): del logs # unused with self._get_writer().as_default(): summary_v2.hparams(self._hparams, trial_id=self._trial_id)
def _write_hparams(hparams): global _writer with _writer.as_default(): hp.hparams(hparams)