Closed haideralimughal closed 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'
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.
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'