search

ctallec / pyvarinf

Python package facilitating the use of Bayesian Deep Learning methods with Variational Inference for PyTorch
MIT License
358 stars 50 forks source link

[FIX] Param naming for torch>.5 #7

Closed diviyank closed 5 years ago

diviyank commented 5 years ago

Bug was raising the following error:

Traceback (most recent call last):
  File "/home/diviyan/conda/lib/python3.6/site-packages/pyvarinf/vi.py", line 232, in __init__
    learn_mean, learn_rho)
  File "/home/diviyan/conda/lib/python3.6/site-packages/pyvarinf/vi.py", line 274, in _variationalize_module
    learn_mean, learn_rho)
  File "/home/diviyan/conda/lib/python3.6/site-packages/pyvarinf/vi.py", line 274, in _variationalize_module
    learn_mean, learn_rho)
  File "/home/diviyan/conda/lib/python3.6/site-packages/pyvarinf/vi.py", line 259, in _variationalize_module
    dico[name].mean)
  File "/home/diviyan/conda/lib/python3.6/site-packages/torch/nn/modules/module.py", line 138, in register_parameter
    raise KeyError("parameter name can't contain \".\"")
KeyError: 'parameter name can\'t contain "."'

Best, Diviyan

diviyank commented 5 years ago

Nvm, was already referred to in #4.