Merge branch 'master' of https://github.com/ctallec/pyvarinf

Author: ctallec

README.md

@@ -171,7 +171,7 @@ var_model.set_prior('mixtgauss', n_mc_samples, sigma_1, sigma_2, pi)
 * `pi` the probability of the first gaussian
 
 # Requirements
-You need to have PyTorch installed for PyVarInf to work (as PyTorch is not readily available on PyPi). To install PyTorch, follow the instructions described [here](http://pytorch.org/#pip-install-pytorch).
+This module requires Python 3. You need to have PyTorch installed for PyVarInf to work (as PyTorch is not readily available on PyPi). To install PyTorch, follow the instructions described [here](http://pytorch.org/#pip-install-pytorch).
 
 # References
 * [1] Blundell, Charles, Cornebise, Julien, Kavukcuoglu, Koray, and Wierstra, Daan. Weight Uncertainty in Neural Networks. In *International Conference on Machine Learning*, pp. 1613–1622, 2015.

pyvarinf/vi.py

@@ -255,10 +255,10 @@ def _variationalize_module(self, dico, module, prefix, zero_mean,
                     None)
 
                 if learn_mean:
-                    self.register_parameter(prefix + '.' + name + '_mean',
+                    self.register_parameter(prefix + '_' + name + '_mean',
                                             dico[name].mean)
                 if learn_rho:
-                    self.register_parameter(prefix + '.' + name + '_rho',
+                    self.register_parameter(prefix + '_' + name + '_rho',
                                             dico[name].rho)
 
             to_erase.append(name)
@@ -269,7 +269,7 @@ def _variationalize_module(self, dico, module, prefix, zero_mean,
         for mname, sub_module in module.named_children():
             sub_dico = OrderedDict()
             self._variationalize_module(sub_dico, sub_module,
-                                        prefix + ('.' if prefix else '') +
+                                        prefix + ('_' if prefix else '') +
                                         mname, zero_mean,
                                         learn_mean, learn_rho)
             dico[mname] = sub_dico