fix prompt override model settings in convert mode

This commit is contained in:
iperov 2019-01-11 11:29:18 +04:00
parent 32f0db3dea
commit e6fa021c3d

View file

@ -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.")