-
Feature request for AMP support in VQ-VAE training.
So far, I tried naively modifying the `train` function in `train_vqvae.py` like so:
```python
# ...
for i, (img, label) in enumerate(loader):…
-
Thank you for sharing this great code.
I have used infovae as a subsitute of beta-vae and traditional vae (beta=1).
However, I think that your vq-vae-2 better reconstructs images.
Unfortunately…
kimdn updated
4 years ago
-
I found `ctx.needs_input_grad[1]` is `False` during training VQ-VAE. Is this correct, and does it mean the embedding of the codebook does not update during training?
https://github.com/ritheshkumar…
zhxgj updated
4 months ago
-
I ran
`python /people/kimd999/script/python/cryoEM/vq-vae-2-pytorch/train_vqvae.py /people/kimd999/MARScryo/dn/data/part/PDX_label_correct/exp_coexp/input --size 256`
It seems to generate sample i…
kimdn updated
4 years ago
-
Hi,
Sorry for bothering but I have following problem when trying to run train_pixelsnail.py on windows. I have succesfully trained the model and I have pretty good sample results from first step. T…
-
I am very interested in your research and plan to follow up on it. Currently only the code for training the VQ-VAE model is available, will the code for FA-VAE be open-sourced?
-
FIrst of all, maybe it is my misunderstanding of the paper, so hope somebody could explain it for me, thanks! :
---
in the paper, the loss is defined as
![Screenshot from 2022-08-30 11-52-26](…
-
Hi,
thanks for your clean implementation ! I was wondering, have you ever tried to calculate the bits / dimension metric (as in the original paper) ? I've tried to do so using the provided code, a…
-
### Feature request
Add support for LlamaGen, an autoregressive image generation model, to the Transformers library. LlamaGen applies the next-token prediction paradigm of large language models to vi…
-
Thanks a lot for this repo!
quick question: once i train on your image set - can i somehow then use the model to estimate the probability that a new image is from the same distribution of the trainin…