Not able to train the PixelCNN about pytorch-vqvae HOT 5 CLOSED

Have you solved it? I got the same question.

from pytorch-vqvae.

No sorry, I ended up just training a VqVae model on my data.

from pytorch-vqvae.

changing n_classes=len(train_dataset._label_encoder) to num_classes=10 would be fine

from pytorch-vqvae.

Hello, thanks for your work.
I am fighting some issues when training your models. I followed your Instructions.

First, I think you may have forgotten to add '--dataset' argument in both of your commands.
Then, I think you forget to import datasets from torchvision in pixelcnn_prior.py.

Eventually, running:

python3 pixelcnn_prior.py --data-folder /tmp/miniimagenet --output-folder models/vqvae --dataset mnist

results in:

AttributeError: 'MNIST' object has no attribute '_label_encoder'

And I have the same issue with CIFAR dataset.

I had the same problem with you in training. The author pixelcnn should only be used for miniimagenet, so import dataset is not needed

from pytorch-vqvae.

I replaced __label_encoder with class_to_idx and it works well. Hope this helps.

from pytorch-vqvae.