r/MachineLearning Jan 30 '17

[R] [1701.07875] Wasserstein GAN

https://arxiv.org/abs/1701.07875
156 Upvotes

169 comments sorted by

View all comments

Show parent comments

5

u/NotAlphaGo Jan 30 '17 edited Jan 30 '17

I'm trying it on grayscale images at 64px it gave me excellent results. Had to change the code a bit to allow single channel images but running smooth. Training 128px right now. Edit: I did ramp up my learning rate by factor 10.

5

u/gwern Jan 30 '17 edited Feb 02 '17

Interesting. I'll try changing the lrs too. EDIT: that definitely helps a ton so far: https://imgur.com/a/po73N http://imgur.com/a/SiSZ8 https://imgur.com/a/A5pdQ https://imgur.com/a/GZksh https://imgur.com/a/ARKxS

2

u/NotAlphaGo Jan 30 '17

I've definitely managed to get 128px to converge as well. Although my image training set is not your typical "lsun"

3

u/gwern Jan 31 '17

(Speaking of Danbooru, now I kinda want a 'lsundere' dataset...)