search

GlassyWing / nvae

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

i am getting this error ModuleNotFoundError: No module named 'robust_loss_pytorch' any suggestions for this ?? #8

Closed haideralimughal closed 3 years ago

haideralimughal commented 3 years ago

Traceback (most recent call last): File "train.py", line 13, in from nvae.vae_celeba import NVAE File "/home/ali/nvae-master/nvae/vae_celeba.py", line 9, in import robust_loss_pytorch ModuleNotFoundError: No module named 'robust_loss_pytorch'

jfrancis71 commented 3 years ago

I have vague memories of coming across this.

Try looking at: https://github.com/jonbarron/robust_loss_pytorch I think I had to install this (follow the installation instructions there).

I hope that helps.