-
I am currently on the second stage to train the pixelsnail model. Do I train the top and bottom models separately, or do I first train the top first and use that checkpoint to train the bottom, since …
-
hello, I am trying to run main-pixelsnail.py and it gives OOM error after executing batches. would you know what could be causing it?
```
`python main-pixelsnail.py latent-data/cifar10_cifar10-stat…
-
Hi
I trained PixelSnail on CIFAR10 and try to sample from the model with this function that I wrote (based on the code from this project notebook) :
```
print("Sample")
…
-
@rosinality
I trained my model from scratch on the LFW face dataset, but the reconstructions using the top and bottom latent maps does not produce significant information.
![1](https://user-images…
-
What command do I run to generated images using your pretrained model?
I tried various methods such as
python3 sample.py --vqvae --top test.jpg
but could not figure it out.
-
Hi!
I try to load the ImageNet dataset to the PixelSnail Model.
after fixing this lines in the pixel_snail.py scripts :
```
`class PixelSNAIL(base.AutoregressiveModel):
"""The PixelSNA…
-
Hi!
In trainer.py the defult settings are for device=torch.device('cpu')).
when I change it to 'cuda' I get this error :
RuntimeError: All input tensors must be on the same device. Received …
-
Hi!
First of all congrats for the nice work.
I am wondering how long it took to train the vq-vae 2 model and the priors, and if you have the pre-trained weights from CheXpert dataset.
Best,
-
Hi!
Is there a way to use the sample function that appears in the notebook within the py scripts / add sampling option to the tensorboard logs?
I tried to convert the imshow to imsave and run i…
-
Hi there, I've been trying to run pixel snail on a custom dataset. Firstly I run:
```python train_vqvae.py ./dataset --size 128```
with a size of 128 as my GPU only has 8GB of memory. Then extra…