From 5ef6840593350c4477e178101b8730619b636410 Mon Sep 17 00:00:00 2001 From: Brigham Lysenko Date: Tue, 13 Aug 2019 19:54:05 -0600 Subject: [PATCH] bug fix --- models/ModelBase.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index 70ad484..b9c98c7 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -198,7 +198,8 @@ class ModelBase(object): self.sort_by_yaw = self.options.get('sort_by_yaw', False) self.random_flip = self.options.get('random_flip', True) if self.batch_cap == 0: - self.batch_cap = self.options['batch_cap'] = self.batch_size + self.options['batch_cap'] = self.batch_size + self.batch_cap = self.options.get('batch_cap',16) self.src_scale_mod = self.options.get('src_scale_mod', 0) if self.src_scale_mod == 0 and 'src_scale_mod' in self.options: