taey16 / pix2pixBEGAN.pytorch

A pytorch implementation of pix2pix + BEGAN (Boundary Equilibrium Generative Adversarial Networks)
183 stars 36 forks source link

Help on Parameter tuning #4

Open kickbox opened 7 years ago

kickbox commented 7 years ago

Thanks for the awesome work! I used pix2pix in tensorflow and had some good success. I tried the same set of images with your model, however the initial training doesn't seem go as expected.

Below is the model measures; capture

Since I have only CPU my training times are typically longer(>3 days on 20 cores).

I have set another run with learning rate of 0.00002 for both D & G. Is there any particular hyper-parameter that needs adjustment?

kickbox commented 7 years ago

Update: All my runs with different values of gamma (.3,.5,.7) and lr(.0002,.00002) produce all black images and the measures look just like above...