stylegan2-pytorch
stylegan2-pytorch copied to clipboard
lucidrains
Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement
I'm having this issue using Torch 1.7.1+cu110. Please see below: ``` (venv) C:\Users\Jerr\PycharmProjects\pythonProject1>stylegan2_pytorch --data C:/Transfer/Downloads/Processed/Compressed/Compressed --network-capacity 256 --trunc-psi 0.5 --aug-prob 0.25 --attn-layers 1 --top-k-training --generate-top-k -frac 0.5 --generate-top-k-gamma 0.99 --no-pl-reg...
Hello, I'm running the training with a directory containing ~1000 images, and it seems to start very well, but after some iterations, it gives a **cuda runtime error, illegal memory...
Hi, thank you so much for making easy repo for styleGAN2. I have a problem with using this code. `stylegan2_pytorch -- DCGAN/Knee_GAN/processed/2/test/ ` When I tried to use the above...
Hello again @lucidrains & StyleGANers! I tried training using `--network-capacity 10 --attn-layers 1 --batch-size 1 gradient-accumulate-every 32 --image-size 512` on a Nvidia 2070 Super with 8 GB of GDDR. The...
Hi, The training was running on with a small GPU so I saved the latest model, 450th epoch (model_450.pt file) and moved it to another machine. I placed the saved...
I noticed that when training with multiple GPUs, sometimes the spawned processes for some reason seem to occupy memory space on the first GPU, i.e. the memory usage is not...
It's mentioned briefly in stylegan2 paper's appendix b, but it is described in [Progressive Growing of GANs](https://arxiv.org/abs/1710.10196) in detail in section 4.1. It's done for both conv and linear layers,...
I can run the model on command line and generate images, but I want to do more with it. Let's say I wanted to add tasks to the model for...
Metadata
Owner
Metadata
Simplest working implementation of Stylegan2, state of the art generative adversarial network, in Pytorch. Enabling everyone to experience disentanglement