vwxyzjn commited on
Commit
dc387d9
1 Parent(s): 1686ac7

pushing model

Browse files
README.md ADDED
@@ -0,0 +1,56 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ tags:
3
+ - BreakoutNoFrameskip-v4
4
+ - deep-reinforcement-learning
5
+ - reinforcement-learning
6
+ - custom-implementation
7
+ model-index:
8
+ - name: DQN
9
+ results:
10
+ - task:
11
+ type: reinforcement-learning
12
+ name: reinforcement-learning
13
+ dataset:
14
+ name: BreakoutNoFrameskip-v4
15
+ type: BreakoutNoFrameskip-v4
16
+ metrics:
17
+ - type: mean_reward
18
+ value: 333.60 +/- 120.61
19
+ name: mean_reward
20
+ verified: false
21
+ ---
22
+
23
+ # (CleanRL) **DQN** Agent Playing **BreakoutNoFrameskip-v4**
24
+
25
+ This is a trained model of a DQN agent playing BreakoutNoFrameskip-v4.
26
+ The model was trained by using [CleanRL](https://github.com/vwxyzjn/cleanrl) and the training code can be
27
+ found [here](https://github.com/vwxyzjn/cleanrl/blob/master/cleanrl/dqn_atari.py).
28
+
29
+
30
+ # Hyperparameters
31
+ ```python
32
+ {'batch_size': 32,
33
+ 'buffer_size': 1000000,
34
+ 'capture_video': False,
35
+ 'cuda': True,
36
+ 'end_e': 0.01,
37
+ 'env_id': 'BreakoutNoFrameskip-v4',
38
+ 'exp_name': 'dqn_atari',
39
+ 'exploration_fraction': 0.1,
40
+ 'gamma': 0.99,
41
+ 'hf_entity': 'cleanrl',
42
+ 'learning_rate': 0.0001,
43
+ 'learning_starts': 80000,
44
+ 'save_model': True,
45
+ 'seed': 1,
46
+ 'start_e': 1,
47
+ 'target_network_frequency': 1000,
48
+ 'torch_deterministic': True,
49
+ 'total_timesteps': 10000000,
50
+ 'track': True,
51
+ 'train_frequency': 4,
52
+ 'upload_model': True,
53
+ 'wandb_entity': None,
54
+ 'wandb_project_name': 'cleanRL'}
55
+ ```
56
+
events.out.tfevents.1666236091.pop-os.943793.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:adf7cf37b6406db5c943c93ffa186160fd7f41c163b530f3a6ff88592a8468af
3
+ size 17240982
events.out.tfevents.1666265159.pop-os.943793.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8a96ec4a35a16ed65f95ae53e0c5ab5d02c2cf07490478516c4cd2b920f6d4b6
3
+ size 618
q_network.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:718999621f9c42b5af6b96f890a13c5d3300c4502c5d14e4afc9d7b31206abac
3
+ size 6747719
replay.mp4 ADDED
Binary file (251 kB). View file
 
videos/BreakoutNoFrameskip-v4__dqn_atari__1__1666236090-eval/rl-video-episode-0.mp4 ADDED
Binary file (340 kB). View file
 
videos/BreakoutNoFrameskip-v4__dqn_atari__1__1666236090-eval/rl-video-episode-1.mp4 ADDED
Binary file (338 kB). View file
 
videos/BreakoutNoFrameskip-v4__dqn_atari__1__1666236090-eval/rl-video-episode-8.mp4 ADDED
Binary file (251 kB). View file