From c6eac6ac4f8ed9dddecee587977b4962fa90113a Mon Sep 17 00:00:00 2001 From: Luke Yeager Date: Thu, 26 May 2016 13:46:30 -0700 Subject: [PATCH] Fix batch accumulation TypeError when unset --- digits/model/forms.py | 1 - digits/model/tasks/caffe_train.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/digits/model/forms.py b/digits/model/forms.py index 276cddc06..e00de8e8f 100644 --- a/digits/model/forms.py +++ b/digits/model/forms.py @@ -129,7 +129,6 @@ def validate_py_ext(form, field): ) batch_accumulation = utils.forms.IntegerField('Batch Accumulation', - default=1, validators = [ validators.NumberRange(min=1), validators.Optional(), diff --git a/digits/model/tasks/caffe_train.py b/digits/model/tasks/caffe_train.py index 58ea0cfa0..b048f13bd 100644 --- a/digits/model/tasks/caffe_train.py +++ b/digits/model/tasks/caffe_train.py @@ -646,7 +646,7 @@ def save_files_generic(self): # Batch accumulation from digits.frameworks import CaffeFramework - if CaffeFramework().can_accumulate_gradients(): + if self.batch_accumulation and CaffeFramework().can_accumulate_gradients(): solver.iter_size = self.batch_accumulation # Epochs -> Iterations