Adjust model checkpointing to effective epochs regime

Author: ASvyatkovskiy

plasma/models/builder.py

@@ -124,7 +124,7 @@ def load_model_weights(self,model):
 			print('no previous checkpoint found')
 			return -1
 		else:
-			max_epoch = min(self.conf['training']['num_epochs']-1,max(epochs))
+                        max_epoch = max(epochs)
 			print('loading from epoch {}'.format(max_epoch))
 			model.load_weights(self.get_save_path(max_epoch))
 			return max_epoch
@@ -135,7 +135,7 @@ def get_latest_save_path(self):
 			print('no previous checkpoint found')
 			return ''
 		else:
-			max_epoch = min(self.conf['training']['num_epochs'],max(epochs))
+                        max_epoch = max(epochs)
 			print('loading from epoch {}'.format(max_epoch))
 			return self.get_save_path(max_epoch)