Merge pull request #1063 from dmitriy-serdyuk/libgpuarray

Refactor to use with libgpuarray
mila-iqia · Apr 19, 2016 · 1c90f63 · 1c90f63
2 parents 8efd188 + 97f22fc
commit 1c90f63
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/blocks/monitoring/aggregation.py b/blocks/monitoring/aggregation.py
@@ -124,12 +124,12 @@ def get_aggregator(self):
                                    tensor.zeros_like(numerator_acc)),
                                   (denominator_acc,
                                    tensor.zeros_like(denominator_acc)),
-                                  (initialized, 0.)]
+                                  (initialized, tensor.zeros_like(initialized))]
         accumulation_updates = [(numerator_acc,
                                  conditional_update_num),
                                 (denominator_acc,
                                  conditional_update_den),
-                                (initialized, 1.)]
+                                (initialized, tensor.ones_like(initialized))]
         aggregator = Aggregator(aggregation_scheme=self,
                                 initialization_updates=initialization_updates,
                                 accumulation_updates=accumulation_updates,