AttributeError: type object 'HyperboloidWrappedNormal' has no attribute 'loc'

[chenyv118]

Thanks for your work, but I have a problem when trying your code, the code is currently not working, specifically executing the command:

python main.py --dataset=mnist --batch_size=100 --epochs=100 --model=hyperbolic --wandb --namestr="MNIST 2-HyperbolicVAE"

Problem AttributeError: type object 'HyperboloidWrappedNormal' has no attribute 'loc' Is there any configuration in the code that I haven't completed, or is it caused by the wrong environment? The specific stack information is shown in the following figure:

[chenyv118]

I added these two statements here, first initialize loc and scale, it seems that the code can run, but I am not sure if it can solve the problem, for example, there may be differences in the algorithm

[joeybose]

Hi,

This does seem to be a version issue. I got the code to run again if you simply comment out the arg_constraints dictionary in the class. Specifically, just comment out this dictionary:

https://github.com/joeybose/HyperbolicNF/blob/master/distributions/wrapped_normal.py#L11