Esempio n. 1
0
 def create_writer(self):
     """
     ## Create TensorFlow summary writer
     """
     self.logger.add_writer(
         tensorboard_writer.Writer(
             tf.compat.v1.summary.FileWriter(str(self.info.summary_path))))
Esempio n. 2
0
 def create_writer(self, session: tf.Session):
     """
     ## Create TensorFlow summary writer
     """
     self.logger.add_writer(
         tensorboard_writer.Writer(
             tf.summary.FileWriter(str(self.info.summary_path),
                                   session.graph)))