b263ab3a2904ee85910623cd770eb6763d81e4a9,niftynet/engine/event_tensorboard.py,TensorBoardLogger,__init__,#TensorBoardLogger#Any#Any#Any#Any#Any#,19

Before Change

                 outputs_collector,
                 is_training,
                 **_unused):
        if not is_training:
            return
        self.tensorboard_every_n = tensorboard_every_n
        self.summary_dir = summary_dir
        // the collector provides TF summary ops
        self.outputs_collector = outputs_collector
        // initialise summary writer
        self.writer_train = tf.summary.FileWriter(
            os.path.join(self.summary_dir, TRAIN), graph)
        self.writer_valid = tf.summary.FileWriter(
            os.path.join(self.summary_dir, VALID), graph)
        ITER_STARTED.connect(self.read_tensorboard_op)
        ITER_FINISHED.connect(self.write_tensorboard)

After Change

        self.outputs_collector = outputs_collector
        // initialise summary writer
        self.writer_train = tf.summary.FileWriter(
            os.path.join(self.summary_dir, TRAIN), tf.get_default_graph())
        self.writer_valid = tf.summary.FileWriter(
            os.path.join(self.summary_dir, VALID), tf.get_default_graph())

        ITER_STARTED.connect(self.read_tensorboard_op)
        ITER_FINISHED.connect(self.write_tensorboard)

Instances