From 3a2bc5ed0e71336a1b7e822d46555b8e32c58eb6 Mon Sep 17 00:00:00 2001
From: Brigham Lysenko <brighamlysenko@gmail.com>
Date: Wed, 14 Aug 2019 16:36:19 -0600
Subject: [PATCH] added safeguards

---
 models/ModelBase.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/models/ModelBase.py b/models/ModelBase.py
index 756bff9..66b3090 100644
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -566,8 +566,10 @@ class ModelBase(object):
             if self.batch_size == 1:
                 self.paddle = 'ping'
             if self.paddle == 'ping':
+                self.save()
                 self.set_batch_size(self.batch_size + 1)
             else:
+                self.save()
                 self.set_batch_size(self.batch_size - 1)
 
         self.iter += 1