Fixed epoch offset of summaries

src/utils/trainer.py

@@ -328,7 +328,7 @@ class ModelTrainer():
                     break
 
             # Write train summaries
-            self.write_summaries(epoch, 'train')
+            self.write_summaries(epoch+1, 'train')
             
             if self.validation_data_generator != None:
                 for validation_xy in self.validation_data_generator:
@@ -343,7 +343,7 @@ class ModelTrainer():
                     self.update_summaries(losses,outputs,validation_y,"val")
                         
                 # Write validation summaries
-                self.write_summaries(epoch,"val")
+                self.write_summaries(epoch+1,"val")
             
             if self.test_data_generator != None:
             
@@ -361,7 +361,7 @@ class ModelTrainer():
                     self.update_summaries(losses,outputs,test_y,"test")
 
                 # Write test summaries
-                self.write_summaries(epoch,"test")
+                self.write_summaries(epoch+1,"test")
 
             
             if self.scalar_summaries != None:
@@ -374,4 +374,4 @@ class ModelTrainer():
             if self.scalar_summaries != None:
                 self.scalar_summaries.reset_summaries()
             #Save the model weights
-            self.save_model(epoch)
+            self.save_model(epoch+1)