Fix

tingiskhan · Mar 17, 2020 · 3185b08 · 3185b08
1 parent 0d14168
commit 3185b08
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pyfilter/__init__.py b/pyfilter/__init__.py
@@ -1 +1 @@
-__version__ = '0.6.9'
+__version__ = '0.6.10'
diff --git a/pyfilter/inference/varapprox/meanfield.py b/pyfilter/inference/varapprox/meanfield.py
@@ -41,10 +41,10 @@ def get_parameters(self):
         return self._mean, self._std
 
     def initialize(self, parameters, *args):
-        self._mean = torch.zeros(sum(p.c_numel() for p in parameters))
-        self._std = torch.ones_like(self._mean)
+        stacked = stacker(parameters, lambda u: u.t_values)
 
-        stacked = stacker(parameters)
+        self._mean = torch.zeros_like(stacked.concated)
+        self._std = torch.ones_like(self._mean)
 
         for p, msk in zip(parameters, stacked.mask):
             try: