From ea7ec7472156aaaaece29a1d3148a4b216737af8 Mon Sep 17 00:00:00 2001 From: iperov Date: Tue, 23 Apr 2019 18:31:12 +0400 Subject: [PATCH] change help for batch size --- models/ModelBase.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/ModelBase.py b/models/ModelBase.py index c3133e9..9e17aae 100644 --- a/models/ModelBase.py +++ b/models/ModelBase.py @@ -95,7 +95,7 @@ class ModelBase(object): if ask_batch_size and (self.iter == 0 or ask_override): default_batch_size = 0 if self.iter == 0 else self.options.get('batch_size',0) - self.options['batch_size'] = max(0, io.input_int("Batch_size (?:help skip:%d) : " % (default_batch_size), default_batch_size, help_message="Larger batch size is always better for NN's generalization, but it can cause Out of Memory error. Tune this value for your videocard manually.")) + self.options['batch_size'] = max(0, io.input_int("Batch_size (?:help skip:%d) : " % (default_batch_size), default_batch_size, help_message="Larger batch size is better for NN's generalization, but it can cause Out of Memory error and increases risk of model collapse during training. Tune this value for your videocard manually.")) else: self.options['batch_size'] = self.options.get('batch_size', 0)