-
First of all, thank you for the Implementation.
In your implementation, the output of the PixelSnail is a softmax layer which corresponds to the index of the codebook. I was wondering where the aut…
-
Hey,
I wrapped your implementation of ImageGPT for a project I am doing in collab.
I have single channel images of size 64 with the intention of scaling it to 128.
Running you code crashes (out o…
-
Hi @rosinality,
Thanks for the code. I want to train the model from scratch on LFW dataset and focus on reconstructions only. Will this code do this and also what are the steps and the hyperparameter…
-
Hi! thank you for your good project!
i think your projectis very helpful and beautiful !
Now i am confused by the "attn_mask " in pixelSnail. Besides, i have awared the "attn_mask " also in the pixe…
-
Thanks for your implementation.
Could you tell me some details about the nums of GPUs you used, the training duration on your platform, and the memory occupied?
Thanks!
-
Error message is given below :
```
Traceback (most recent call last):
File "train_pixelsnail.py", line 138, in
train(args, i, loader, model, optimizer, scheduler, device)
File "train_pi…
Corea updated
5 years ago
-
Hi,
I've been thinking to utilize VQ-VAE 2 in my project and found that you're already done this great thing! thanks for your implementation.
I've just finished stage 1, followed by code extract…
-
I have 74862 train images and set batch size as 12 with 3 GPUs, this error message appears in the last step of epoch. This issue might be resolved with `drop_last=True` on DataLoader, or modify model …
Corea updated
5 years ago
-
The pixelCNN++ code [here](https://github.com/openai/pixel-cnn) has a "data" folder that this code also seems to rely on (I just copied it from that repo and it seemed to work). That code has imageio…