Commit 59ce73cd authored by Alexander Fuchs's avatar Alexander Fuchs
Browse files

Fixed epoch offset of summaries

parent 283ce360
......@@ -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)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment