google / compare_gan

Compare GAN code.
Apache License 2.0
1.82k stars 319 forks source link

tf-nightly-gpu version < 2 no longer exists #51

Open odakiese opened 4 years ago

odakiese commented 4 years ago

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']