VQ VAE model's reconstruction is a black image

AntixK / PyTorch-VAE

A Collection of Variational Autoencoders (VAE) in PyTorch.

Apache License 2.0

6.44k stars 1.05k forks source link

VQ VAE model's reconstruction is a black image #51

Closed ashishk88 closed 2 years ago

ashishk88 commented 2 years ago

I am using your wonderful library in my research project. There seems to be a bug in the VQ VAE mode class where the reconstruction is a blank image. Is it a known bug? Can you please help me with this issue?

manupillai308 commented 2 years ago

I m also facing the same issue.

jiang-wenbin commented 2 years ago

You should monitor the loss. If the loss grow up dramatically, reduce the learning rate. I set the LR 0.0003

ashishk88 commented 2 years ago

I will try @jiang-wenbin approach. Marking this closed.