--- tags: - PongNoFrameskip-v4 - deep-reinforcement-learning - reinforcement-learning - custom-implementation library_name: cleanrl model-index: - name: DQN results: - task: type: reinforcement-learning name: reinforcement-learning dataset: name: PongNoFrameskip-v4 type: PongNoFrameskip-v4 metrics: - type: mean_reward value: 18.53 +/- 0.00 name: mean_reward verified: false --- # (CleanRL) **DQN** Agent Playing **PongNoFrameskip-v4** This is a trained model of a DQN agent playing PongNoFrameskip-v4. The model was trained by using [CleanRL](https://github.com/vwxyzjn/cleanrl) and the most up-to-date training code can be found [here](https://github.com/vwxyzjn/cleanrl/blob/master/cleanrl/DDQN.py). ## Get Started To use this model, please install the `cleanrl` package with the following command: ``` pip install "cleanrl[DDQN]" python -m cleanrl_utils.enjoy --exp-name DDQN --env-id PongNoFrameskip-v4 ``` Please refer to the [documentation](https://docs.cleanrl.dev/get-started/zoo/) for more detail. ## Command to reproduce the training ```bash curl -OL https://huggingface.co/pfunk/PongNoFrameskip-v4-DDQN-seed3/raw/main/dqn_atari.py curl -OL https://huggingface.co/pfunk/PongNoFrameskip-v4-DDQN-seed3/raw/main/pyproject.toml curl -OL https://huggingface.co/pfunk/PongNoFrameskip-v4-DDQN-seed3/raw/main/poetry.lock poetry install --all-extras python dqn_atari.py --track --wandb-entity pfunk --wandb-project-name dqpn --capture-video true --save-model true --upload-model true --hf-entity pfunk --exp-name DDQN --target-network-frequency 1000 --seed 3 --double-learning ``` # Hyperparameters ```python {'alg_type': 'dqn_atari.py', 'batch_size': 32, 'buffer_size': 1000000, 'capture_video': True, 'cuda': True, 'double_learning': True, 'end_e': 0.05, 'env_id': 'PongNoFrameskip-v4', 'exp_name': 'DDQN', 'exploration_fraction': 0.2, 'gamma': 0.99, 'hf_entity': 'pfunk', 'learning_rate': 0.0001, 'learning_starts': 10000, 'max_gradient_norm': inf, 'save_model': True, 'seed': 3, 'start_e': 1.0, 'target_network_frequency': 1000, 'target_tau': 1.0, 'torch_deterministic': True, 'total_timesteps': 10000000, 'track': True, 'train_frequency': 1, 'upload_model': True, 'wandb_entity': 'pfunk', 'wandb_project_name': 'dqpn'} ```