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