From 093ca4dba67ac2db543702d6402e7da9028cde39 Mon Sep 17 00:00:00 2001 From: iperov Date: Sun, 21 Apr 2019 21:08:45 +0400 Subject: [PATCH] fix options in ModelBase.py --- models/ModelBase.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index 91b8aa0..83dbdbe 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -117,12 +117,12 @@ class ModelBase(object): else: self.options['src_scale_mod'] = self.options.get('src_scale_mod', 0) - self.write_preview_history = self.options['write_preview_history'] - if not self.options['write_preview_history']: + self.write_preview_history = self.options.get('write_preview_history', False) + if not self.write_preview_history and 'write_preview_history' in self.options: self.options.pop('write_preview_history') - self.target_iter = self.options['target_iter'] - if self.options['target_iter'] == 0: + self.target_iter = self.options.get('target_iter',0) + 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)