Fix batch accumulation TypeError when unset

NVIDIA · May 26, 2016 · c6eac6a · c6eac6a
1 parent b84c569
commit c6eac6a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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