yuanming-hu / exposure

Learning infinite-resolution image processing with GAN and RL from unpaired image datasets, using a differentiable photo editing model.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

WGAN error

rlica opened this issue · comments

Any idea if this is related to the new version of tensorflow?

tensorflow 1.0.1
tensorflow-gpu 1.10.0

** WGAN
routine_loss (?, 1)
pg_loss (?, 1)
Traceback (most recent call last):
File "evaluate.py", line 35, in
evaluate()
File "evaluate.py", line 27, in evaluate
net = GAN(cfg, restore=True)
File "/home/licarazvan90/exposure/net.py", line 174, in init
alpha_dist = tf.contrib.distributions.Uniform(low=0., high=1.)
TypeError: init() got an unexpected keyword argument 'low'

The code should work with tensorflow-gpu 1.10.0. Could you remove uninstall tensorflow 1.0.1 and try again? Just having the GPU version installed should be sufficient.

Thanks, indeed it works! Throught pip I removed tensorflow and still got some errors and then I changed to conda and everything worked fine. So it was some environment problem (note for others having similar errors).
Cheers!