TF Version must < 2.0?

[weihongwei0586]

The code cant work in my environment where TF version is the newest(2.2.0).
because the tf.contrib moudle has been removed?

[agarwl]

Yeah, I think Dopamine library still uses tensorflow 1.x and the code is only compatible with that for now :/ . @psc-g for visibility about dopamine.

[psc-g]

we're working on both a jax version of dopamine and a TF2 compliant version, stay tuned!

[weihongwei0586]

we're working on both a jax version of dopamine and a TF2 compliant version, stay tuned!

Thank you very much!

[agarwl]

Dopamine now supports JAX agents! The main difference in offline agents is to simply use the FixedReplayBuffer (which directly reads data from numpy) instead of the tensorflow buffer. For an example, see the implementation of QR-DQN in JAX.