google / dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

Home Page:https://github.com/google/dopamine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

the return value of step function in atari

lucasliunju opened this issue · comments

Hi, I try to run the code with default setting for aatari (Pong). I find the step function (dopamine/dopamine/discrete_domains/atari_lib.py +467) returns five values but the code just defines four values. May I ask how to modify that?

In addition, I find the step function returns 2 values about "done" and I think this maybe the main reason.

Thank you very much!

Hi @lucasliunju, Dopamine doesn’t support Gym version 0.26. Breaking changes were introduced in v0.26 to the step and reset function (as you pointed out the return signature changed). You should use Gym 0.25.X along with ale-py 0.7.X.

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 \ --base_dir /tmp/dopamine_runs \ --gin_files dopamine/agents/dqn/configs/dqn.gin

Thanks! I also had this issue, and it now works after manually running pip install ale-py==0.7.5 gym==0.25.2

Can you update the requirements.txt file to specify the versions that dopamine is currently compatible with?

(I followed the instructions which say pip install -r dopamine/requirements.txt, but that ends up with the newer incompatible versions of ale-py and gym, so it currently doesn't work.)

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

I would like to ask if is it normal.