Atari model with DQN architecture
Browse files- README.md +37 -0
- atari-test.zip +3 -0
- atari-test/_stable_baselines3_version +1 -0
- atari-test/data +0 -0
- atari-test/policy.optimizer.pth +3 -0
- atari-test/policy.pth +3 -0
- atari-test/pytorch_variables.pth +3 -0
- atari-test/system_info.txt +9 -0
- config.json +0 -0
- results.json +1 -0
README.md
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: stable-baselines3
|
| 3 |
+
tags:
|
| 4 |
+
- SpaceInvadersNoFrameskip-v4
|
| 5 |
+
- deep-reinforcement-learning
|
| 6 |
+
- reinforcement-learning
|
| 7 |
+
- stable-baselines3
|
| 8 |
+
model-index:
|
| 9 |
+
- name: DQN
|
| 10 |
+
results:
|
| 11 |
+
- task:
|
| 12 |
+
type: reinforcement-learning
|
| 13 |
+
name: reinforcement-learning
|
| 14 |
+
dataset:
|
| 15 |
+
name: SpaceInvadersNoFrameskip-v4
|
| 16 |
+
type: SpaceInvadersNoFrameskip-v4
|
| 17 |
+
metrics:
|
| 18 |
+
- type: mean_reward
|
| 19 |
+
value: 720.50 +/- 381.66
|
| 20 |
+
name: mean_reward
|
| 21 |
+
verified: false
|
| 22 |
+
---
|
| 23 |
+
|
| 24 |
+
# **DQN** Agent playing **SpaceInvadersNoFrameskip-v4**
|
| 25 |
+
This is a trained model of a **DQN** agent playing **SpaceInvadersNoFrameskip-v4**
|
| 26 |
+
using the [stable-baselines3 library](https://github.com/DLR-RM/stable-baselines3).
|
| 27 |
+
|
| 28 |
+
## Usage (with Stable-baselines3)
|
| 29 |
+
TODO: Add your code
|
| 30 |
+
|
| 31 |
+
|
| 32 |
+
```python
|
| 33 |
+
from stable_baselines3 import ...
|
| 34 |
+
from huggingface_sb3 import load_from_hub
|
| 35 |
+
|
| 36 |
+
...
|
| 37 |
+
```
|
atari-test.zip
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:6a39cd908e9dd0194eb00f3d9ca38fc8b4c539959ed700e2f50efdd0e87d6146
|
| 3 |
+
size 27482793
|
atari-test/_stable_baselines3_version
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
2.4.1
|
atari-test/data
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
atari-test/policy.optimizer.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:497892f12e908bb7cd6609aec5e076bfd4efea88f64f3dc29b284212f4d0623d
|
| 3 |
+
size 13505852
|
atari-test/policy.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:324c49a01123cae9558989c19f2b9ea31a8e74ff2ba3d338544ef14ac21cb063
|
| 3 |
+
size 13505178
|
atari-test/pytorch_variables.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ebdad4b9cfe9cd22a3abadb5623bf7bb1f6eb2e408740245eb3f2044b0adc018
|
| 3 |
+
size 864
|
atari-test/system_info.txt
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
- OS: macOS-14.6.1-arm64-arm-64bit Darwin Kernel Version 23.6.0: Mon Jul 29 21:16:46 PDT 2024; root:xnu-10063.141.2~1/RELEASE_ARM64_T8112
|
| 2 |
+
- Python: 3.10.16
|
| 3 |
+
- Stable-Baselines3: 2.4.1
|
| 4 |
+
- PyTorch: 2.5.1
|
| 5 |
+
- GPU Enabled: False
|
| 6 |
+
- Numpy: 1.26.4
|
| 7 |
+
- Cloudpickle: 3.1.1
|
| 8 |
+
- Gymnasium: 1.0.0
|
| 9 |
+
- OpenAI Gym: 0.26.2
|
config.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
results.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"mean_reward": 720.5, "std_reward": 381.6572415139008, "is_deterministic": false, "n_eval_episodes": 10, "eval_datetime": "2025-01-28T11:21:47.421199"}
|