search

GlassyWing / nvae

An unofficial toy implementation for NVAE 《A Deep Hierarchical Variational Autoencoder》
Apache License 2.0
108 stars 21 forks source link

Any guidance for this error? #9

Closed haideralimughal closed 3 years ago

haideralimughal commented 3 years ago

I am getting this error when tried to run Python random_sample.py "RuntimeError: Tensors must have the same number of dimensions: got 4 and 2"

GlassyWing commented 3 years ago

I am getting this error when tried to run Python random_sample.py "RuntimeError: Tensors must have the same number of dimensions: got 4 and 2"

Thanks for report, the latent input z need to be changed to this:

line 22:


z = torch.randn((bsz, z_dim, 2, 2))
``