From 9e498c5d71945d6db8b70fef5294950190376325 Mon Sep 17 00:00:00 2001
From: Brigham Lysenko <brighamlysenko@gmail.com>
Date: Tue, 13 Aug 2019 15:43:00 -0600
Subject: [PATCH] added ability to not use ping pong

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

diff --git a/models/ModelBase.py b/models/ModelBase.py
index c6d16d3..639d6a7 100644
--- a/models/ModelBase.py
+++ b/models/ModelBase.py
@@ -526,7 +526,7 @@ class ModelBase(object):
         if self.iter == 1 and self.options.get('ping_pong', True):
             self.set_batch_size(1)
             self.paddle = 'ping'
-        elif not self.options.get('ping_pong', True):
+        elif not self.options.get('ping_pong', True) and self.batch_cap != self.batch_size:
             self.set_batch_size(self.batch_cap)
         sample = self.generate_next_sample()
         iter_time = time.time()