google / compare_gan

Compare GAN code.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

tf-nightly-gpu version < 2 no longer exists

odakiese opened this issue · comments

As tf-nightly-gpu versions below 2 no longer exists on pypi I am using
tensorboard==1.12.2
tensorflow==1.12.0
tensorflow-datasets==1.0.1
tensorflow-estimator==1.14.0
tensorflow-gan==0.0.0.dev0
tensorflow-gpu==1.12.0
tensorflow-hub==0.2.0
tensorflow-metadata==0.21.1

I am able to train using the BigGan network, however when it tries to evaluate I get the error

Traceback (most recent call last):
File "compare_gan/main.py", line 134, in
app.run(main)
File "/home/odak/.conda/envs/tester_venv2/lib/python3.6/site-packages/absl/app.py", line 299, in run
_run_main(main, args)
File "/home/odak/.conda/envs/tester_venv2/lib/python3.6/site-packages/absl/app.py", line 250, in _run_main
sys.exit(main(argv))
File "compare_gan/main.py", line 128, in main
eval_every_steps=FLAGS.eval_every_steps)
File "/home/odak/compare_gan/compare_gan/runner_lib.py", line 349, in run_with_schedule
gan.as_module_spec(),
File "/home/odak/compare_gan/compare_gan/gans/modular_gan.py", line 310, in as_module_spec
self._module_fn, tags_and_args=tags_and_args)
File "/home/odak/.conda/envs/tester_venv2/lib/python3.6/site-packages/tensorflow_hub/native_module.py", line 189, in create_module_spec
if err: raise ValueError(err)
ValueError: A state-holding node x of a module's graph (e.g., a Variable op) must not be subject to a tf.colocate_with(y) constraint unless y is also a state-holding node.
Details: in the graph for tags set(), node 'generator/embed_y/kernel/ExponentialMovingAverage' has op 'VarHandleOp', which counts as state-holding, but Operation.colocation_groups() == [b'loc:@generator/embed_y/kernel/ExponentialMovingAverage/Read/ReadVariableOp']