ITADN

It seems the `cdf()` in `DiagNormal` has not been implemented?

#40OpenBill-Gots 创建于 2024-02-23
bug
B
Bill-Gotscommented
Running the following lines ``` import zuko model = zuko.flows.GF(3) print(model.base().cdf(0.)) ``` I got a `NotImplementedError` in `torch\distributions\distribution.py`. Plus, I wanted to truncate the `base` in `flows.GF`, but I failed to manage it. ``` import zuko model = zuko.flows.GF(3) model.base = zuko.distributions.Truncated(model.base(), -1., 1.) samples = model().rsample((32,)) print(samples) ``` Running the code above I got ``` File "D:\Program Files\Python3.11.4\Lib\site-packages\zuko\distributions.py", line 458, in __init__ assert not base.event_shape, "'base' has to be univariate" AssertionError: 'base' has to be univariate ``` I am new in normalization flows so I don't know how to correct this. Any help is appreciated.
3 条评论