问题标签 [pyro.ai]

For questions regarding programming in ECMAScript (JavaScript/JS) and its various dialects/implementations (excluding ActionScript). Note JavaScript is NOT the same as Java! Please include all relevant tags on your question; e.g., [node.js], [jquery], [json], [reactjs], [angular], [ember.js], [vue.js], [typescript], [svelte], etc.

0 投票
0 回答
12 浏览

pyro.ai - 结合重新参数化和自动枚举时的问题

我正在尝试在 pyro 中复制numpyro 中的annotators.py示例。

更详细地说,当与 pyro 中离散变量的自动枚举相结合时,重新参数化会产生问题。

下面的代码与 numpyro 示例完全相同,除了从 jax 到 torch 的明显转换。

在使用 NUTS 的第一次 MCMC 迭代中(遵循 numpyro 示例),我得到以下调试打印

在第二次迭代中,c枚举时,我得到以下调试打印

令我困惑的是betasize 从(5, 4, 4)变为的事实(4, 4)。当我删除重新参数化时,这不会发生。

关于在哪里寻找以了解会发生什么的任何建议?

非常感谢您抽出宝贵时间。

最好的,彼得罗

也发布在 pyro 论坛上:问题