Comments (11)
By loading the CelebA HQ dataset and using the same code, I'm getting similar results to NVIDIA.
Training time is 24 days on Titan X (Pascal) GPU.
The model works fine and the sample result that generated at 512X512 resolution is attached.
from pggan-pytorch.
At the moment, the dataloader resize the input image on cpu in real time. If resize all images in advance, the training time would be decreased significantly. By the way, the result now looks pretty good :)
Are you preparing paper using pggan? just curious :)
from pggan-pytorch.
yeah, I guess it's been a quite long time since my last update.
The code is working properly until it reaches 256 resolution.
I have not tried with CelebA-HQ yet, but it should be working as well :)
Most bugs have been fixed already, so please feel free to use this.
(I will update the 256x256 results soon. too busy these days!)
from pggan-pytorch.
Am I correct in saying this is still very slow? I've had it running for 3 days on a Titan XP, and my resulting faces are somewhat... monstrous. I'm using default values, btw.
from pggan-pytorch.
my results are shown in README.
what do you mean monstrous? Is it far from mine?
could you show me what you've got?
And yes, the training is slow. The authors trained almost a week as well.
from pggan-pytorch.
Well, some of the images are pretty good, but even after almost a week, many of the images come out terrible -- pieces don't fit together, or are really smudged. Did NVIDIA hand-pick the images in the paper to look that good?
from pggan-pytorch.
I've trained to reproduce the issue with default settings.
The current result is like below, and it took around 3 days using P40 single GPU.
Yes, I see some images look smashed terribly. (but not that much, though.)
My hypothesis about this is that:
-
- NVIDIA authors hand-picked good looking images? (not sure)
-
- It might be working well if using easier domain? (performing face detection and crop on CelebA, or using Anime dataset?)
-
- There is something wrong in my code? (possibly not... checked more than thousands of times)
-
- Wrong optimization parameter? (the result changes significantly depending on the adam optimizer beta1/beta2 values.)
I think we need to go through this issue based on the hypothesis above.
what do you think?
my result so far.
from pggan-pytorch.
Yeah I'm getting the same kind of results. Some areas are 'smushed', but more importantly, the network doesn't seem to be able to combine the pieces from the features in a seamless way, causing the uncanny valley / 'Frankenstein' effect. This problem was noticeable at lower resolution levels as well, but I was hoping it would get better, though it didn't.
Maybe the algorithm needs to run longer on lower resolutions to work out those kinks before we go up in resolution. Perhaps once those patterns are there, they can't be eliminated, since the lower layers have already been formed for the most part.
from pggan-pytorch.
I didn't realize you weren't supporting Wasserstein loss yet. This could be a very big element.
from pggan-pytorch.
Have you tried to use wasserstein loss by any chance?
from pggan-pytorch.
Thank you for sharing this code in pytorch!
I have cited this GitHub in my thesis.
Pggan experiments are a part of my thesis work, have not planned more than this for now:-)
from pggan-pytorch.
Related Issues (20)
- how to load pretrained generator ? HOT 1
- Why even use PyTorch's DataLoader in dataloader.py HOT 4
- Trying to understand the training on LSUN dataset with multi-class labels HOT 1
- about multi-gpu
- weight_scale for equalized_learning
- Resuming Training HOT 1
- Is config.smoothing ever used?
- Error in trainer.py with indentation HOT 5
- Which version of Pytorch should we use ? HOT 1
- I met come errors in trainer.py HOT 5
- question是 HOT 1
- KeyError: 'fadein_block' HOT 2
- Questions about PyTorch version HOT 3
- python3.6
- so much errors in code
- AttributeError: 'DataParallel' object has no attribute 'grow_network' HOT 2
- How to train a custom dataset of 4:3 800x600 image dataset?
- Problem when training on my dataset HOT 1
- why in the deconv function you called Equalized_conv instead of Equalized_deconv instead ??? HOT 1
- Multi-GPU training error
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from pggan-pytorch.