diff --git a/models/ModelBase.py b/models/ModelBase.py index 379d305..453a462 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -72,7 +72,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.is_training_mode and self.iter != 0 and io.input_in_time ("Press enter in 2 seconds to override model settings.", 2) + ask_override = self.is_training_mode and self.iter != 0 and io.input_in_time ("Press enter in 2 seconds to override model settings.", 5 if io.is_colab() else 2 ) yn_str = {True:'y',False:'n'}