Views
No views yet
cleanrl package with the following command:pip install "cleanrl[c51_atari_jax]"
python -m cleanrl_utils.enjoy --exp-name c51_atari_jax --env-id PongNoFrameskip-v41curl -OL https://huggingface.co/kinalmehta/PongNoFrameskip-v4-c51_atari_jax-seed1/raw/main/c51_atari_jax.py
2curl -OL https://huggingface.co/kinalmehta/PongNoFrameskip-v4-c51_atari_jax-seed1/raw/main/pyproject.toml
3curl -OL https://huggingface.co/kinalmehta/PongNoFrameskip-v4-c51_atari_jax-seed1/raw/main/poetry.lock
4poetry install --all-extras
5python c51_atari_jax.py --save-model --upload-model --hf-entity kinalmehta --env-id PongNoFrameskip-v41{'batch_size': 32,
2 'buffer_size': 1000000,
3 'capture_video': False,
4 'end_e': 0.01,
5 'env_id': 'PongNoFrameskip-v4',
6 'exp_name': 'c51_atari_jax',
7 'exploration_fraction': 0.1,
8 'gamma': 0.99,
9 'hf_entity': 'kinalmehta',
10 'learning_rate': 0.00025,
11 'learning_starts': 80000,
12 'n_atoms': 51,
13 'save_model': True,
14 'seed': 1,
15 'start_e': 1,
16 'target_network_frequency': 10000,
17 'total_timesteps': 10000000,
18 'track': False,
19 'train_frequency': 4,
20 'upload_model': True,
21 'v_max': 10,
22 'v_min': -10,
23 'wandb_entity': None,
24 'wandb_project_name': 'cleanRL'}