MishaLaskin / vqvae

A pytorch implementation of the vector quantized variational autoencoder (https://arxiv.org/abs/1711.00937)
601 stars 71 forks source link

Not able to find pretrained checkpoint #17

Closed MeruguRanjith closed 23 hours ago

MeruguRanjith commented 23 hours ago

Could you share us any trained checkpoint for reference? And also, what is the ideal range of your codebook from the trained checkpoint.

When we train this Pytorch code with L1 and perceptual loss, the codebook value range as drastically gone up to -25 to 25.