vwxyzjn commited on
Commit
cb8ace8
1 Parent(s): c218333

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: 2.70 +/- 4.12
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': '',
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': 10000,
50
+ 'track': False,
51
+ 'train_frequency': 4,
52
+ 'upload_model': True,
53
+ 'wandb_entity': None,
54
+ 'wandb_project_name': 'cleanRL'}
55
+ ```
56
+
events.out.tfevents.1666226068.pop-os.942086.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc10b883c4dce0532ad995a1feb69d34036ce0e475473e2620151e1a03bee35e
3
+ size 9004
events.out.tfevents.1666226092.pop-os.942086.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a0e79e93fb645a2c9191974d8b204a4aff91d1fd43dd04d23d37001fd023de0
3
+ size 618
q_network.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1636d410bb9db11837973f98f78023bb8f5008654fedda3014e1b11c6b1c552f
3
+ size 6747719
replay.mp4 ADDED
Binary file (21.9 kB). View file
 
videos/BreakoutNoFrameskip-v4__dqn_atari__1__1666226068-eval/rl-video-episode-0.mp4 ADDED
Binary file (22.3 kB). View file
 
videos/BreakoutNoFrameskip-v4__dqn_atari__1__1666226068-eval/rl-video-episode-1.mp4 ADDED
Binary file (22.4 kB). View file
 
videos/BreakoutNoFrameskip-v4__dqn_atari__1__1666226068-eval/rl-video-episode-8.mp4 ADDED
Binary file (21.9 kB). View file