From e6fa021c3d57497e91419a1a6657ce81acd068ea Mon Sep 17 00:00:00 2001 From: iperov Date: Fri, 11 Jan 2019 11:29:18 +0400 Subject: [PATCH] fix prompt override model settings in convert mode --- models/ModelBase.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index 8e746a9..dcad4bf 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -49,7 +49,7 @@ class ModelBase(object): self.loss_history = model_data['loss_history'] if 'loss_history' in model_data.keys() else [] self.sample_for_preview = model_data['sample_for_preview'] if 'sample_for_preview' in model_data.keys() else None - ask_override = self.epoch != 0 and input_in_time ("Press enter during 2 seconds to override some model settings.", 2) + ask_override = self.is_training_mode and self.epoch != 0 and input_in_time ("Press enter during 2 seconds to override some model settings.", 2) if self.epoch == 0: print ("\nModel first run. Enter model options as default for each run.")