lucasliunju

I suffer from the same problem and I want to ask whether you solve it. Thanks.

Thank you very much! May I ask if I can also run the code in TPU-VM with JAX? Best, Lucas

Thank you very much! I'll have a try.

Thank you very much! May I ask how to train the DQN agent with JAX. I follow this command and I find that is using tensorflow: `python -um dopamine.discrete_domains.train \...

Thanks for your update. I try to run the code of jax-based dqn in Atari (Pong) and I find the return is less than the result in this link: https://google.github.io/dopamine/baselines/atari/plots.html...

Hi Yang, Thanks so much for your help! Yong

Hi Yang, I still have a question about how to connect jax and multi-host tpu since I have not find this part on the code. My current code is :...

Hello! I'm very sorry to trouble you! I want to ask whether the mode (tmux, child) make a great influence in training. Because I have trouble in tmux and I...

I am suffering from the same problem and I don't know what should I do.

Dear agarwl, I try to follow your provided code and reproduce the results of offline dqn based on jax. I find the training speed of jax is quite slow compared...