From 1707e54c302e27dd527f59ac65a65d86d0d6e7d6 Mon Sep 17 00:00:00 2001 From: Brigham Lysenko Date: Tue, 13 Aug 2019 19:56:56 -0600 Subject: [PATCH] bug fix --- models/ModelBase.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index b9c98c7..4499773 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -152,7 +152,7 @@ class ModelBase(object): self.options['paddle'] = self.options.get('paddle','ping') else: - self.options['batch_cap'] = self.options.get('batch_cap', 0) + self.options['batch_cap'] = self.options.get('batch_cap', 16) self.options['ping_pong'] = self.options.get('ping_pong', True) if ask_sort_by_yaw: @@ -193,7 +193,7 @@ class ModelBase(object): if self.target_iter == 0 and 'target_iter' in self.options: self.options.pop('target_iter') - self.batch_size = self.options.get('batch_size', 0) + self.batch_size = self.options.get('batch_size', 8) self.batch_cap = self.options.get('batch_cap',16) self.sort_by_yaw = self.options.get('sort_by_yaw', False) self.random_flip = self.options.get('random_flip', True)