log last epoch

mr-francois · Aug 29, 2022 · d0c92b9 · d0c92b9
1 parent a56a718
commit d0c92b9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cmtb/confusion_matrix_tensorboard.py b/cmtb/confusion_matrix_tensorboard.py
@@ -103,11 +103,11 @@ def on_epoch_begin(self, epoch, logs):
             tf.summary.image(name="confusion matrix train", data=self.train_images, step=int(epoch - 1))
             tf.summary.image(name="confusion matrix validation", data=self.val_images, step=int(epoch - 1))
             file_writer.flush()
-            self.epoch < - epoch
+            self.epoch = epoch
 
     def on_train_end(self, logs):
 
-        epoch = self.epoch + 1
+        epoch = self.epoch
         cm_train = np.load(self.cm_dir + "/cm_train_" + str(epoch - 1) + ".npy")
 
         for i in range(len(self.model.metrics)):