loweegee commited on
Commit
5bf1d89
·
verified ·
1 Parent(s): 47a66af

Initial commit

Browse files
README.md CHANGED
@@ -16,7 +16,7 @@ model-index:
16
  type: SpaceInvadersNoFrameskip-v4
17
  metrics:
18
  - type: mean_reward
19
- value: 5.00 +/- 7.07
20
  name: mean_reward
21
  verified: false
22
  ---
@@ -63,7 +63,7 @@ python -m rl_zoo3.push_to_hub --algo dqn --env SpaceInvadersNoFrameskip-v4 -f lo
63
 
64
  ## Hyperparameters
65
  ```python
66
- OrderedDict([('batch_size', 16),
67
  ('buffer_size', 100000),
68
  ('env_wrapper',
69
  ['stable_baselines3.common.atari_wrappers.AtariWrapper']),
@@ -73,7 +73,7 @@ OrderedDict([('batch_size', 16),
73
  ('gradient_steps', 1),
74
  ('learning_rate', 0.0001),
75
  ('learning_starts', 100000),
76
- ('n_timesteps', 100000.0),
77
  ('optimize_memory_usage', False),
78
  ('policy', 'CnnPolicy'),
79
  ('target_update_interval', 1000),
 
16
  type: SpaceInvadersNoFrameskip-v4
17
  metrics:
18
  - type: mean_reward
19
+ value: 680.50 +/- 265.86
20
  name: mean_reward
21
  verified: false
22
  ---
 
63
 
64
  ## Hyperparameters
65
  ```python
66
+ OrderedDict([('batch_size', 32),
67
  ('buffer_size', 100000),
68
  ('env_wrapper',
69
  ['stable_baselines3.common.atari_wrappers.AtariWrapper']),
 
73
  ('gradient_steps', 1),
74
  ('learning_rate', 0.0001),
75
  ('learning_starts', 100000),
76
+ ('n_timesteps', 1000000.0),
77
  ('optimize_memory_usage', False),
78
  ('policy', 'CnnPolicy'),
79
  ('target_update_interval', 1000),
args.yml CHANGED
@@ -56,7 +56,7 @@
56
  - - save_replay_buffer
57
  - false
58
  - - seed
59
- - 19963506
60
  - - storage
61
  - null
62
  - - study_name
 
56
  - - save_replay_buffer
57
  - false
58
  - - seed
59
+ - 4201572127
60
  - - storage
61
  - null
62
  - - study_name
config.yml CHANGED
@@ -1,6 +1,6 @@
1
  !!python/object/apply:collections.OrderedDict
2
  - - - batch_size
3
- - 16
4
  - - buffer_size
5
  - 100000
6
  - - env_wrapper
@@ -18,7 +18,7 @@
18
  - - learning_starts
19
  - 100000
20
  - - n_timesteps
21
- - 100000.0
22
  - - optimize_memory_usage
23
  - false
24
  - - policy
 
1
  !!python/object/apply:collections.OrderedDict
2
  - - - batch_size
3
+ - 32
4
  - - buffer_size
5
  - 100000
6
  - - env_wrapper
 
18
  - - learning_starts
19
  - 100000
20
  - - n_timesteps
21
+ - 1000000.0
22
  - - optimize_memory_usage
23
  - false
24
  - - policy
dqn-SpaceInvadersNoFrameskip-v4.zip CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a46777f02d7924d9af7ac98c8eb9fb34996a54fd79768c0abb2445fdf46ecda4
3
- size 13713174
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8544187546799e12555ed4c9b365598aaf990457429b8744befc8f2064ec82be
3
+ size 27218426
dqn-SpaceInvadersNoFrameskip-v4/data CHANGED
The diff for this file is too large to render. See raw diff
 
dqn-SpaceInvadersNoFrameskip-v4/policy.optimizer.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:215ea7d8898faa9284464c7109532cf36390d330fccab8d77eeba20628a32876
3
- size 1120
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fd9a562cebfe13f38a07d4dcfa28979ef1905988d5e5df752db128fc883c9039
3
+ size 13506172
dqn-SpaceInvadersNoFrameskip-v4/policy.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:89a1de80e14a7e4816072062ee3cff26f7cb36b3db103b294b2b845c495a2bae
3
- size 13505178
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9fdabbc604dead30b5b2f477eac0a970746589f7868d481395271fff9c9c959
3
+ size 13505370
dqn-SpaceInvadersNoFrameskip-v4/system_info.txt CHANGED
@@ -2,7 +2,7 @@
2
  - Python: 3.11.13
3
  - Stable-Baselines3: 2.7.0a1
4
  - PyTorch: 2.6.0+cu124
5
- - GPU Enabled: False
6
  - Numpy: 2.0.2
7
  - Cloudpickle: 3.1.1
8
  - Gymnasium: 1.1.1
 
2
  - Python: 3.11.13
3
  - Stable-Baselines3: 2.7.0a1
4
  - PyTorch: 2.6.0+cu124
5
+ - GPU Enabled: True
6
  - Numpy: 2.0.2
7
  - Cloudpickle: 3.1.1
8
  - Gymnasium: 1.1.1
results.json CHANGED
@@ -1 +1 @@
1
- {"mean_reward": 5.0, "std_reward": 7.0710678118654755, "is_deterministic": false, "n_eval_episodes": 10, "eval_datetime": "2025-07-21T11:19:54.633865"}
 
1
+ {"mean_reward": 680.5, "std_reward": 265.8613360381686, "is_deterministic": false, "n_eval_episodes": 10, "eval_datetime": "2025-07-21T15:13:33.190261"}
train_eval_metrics.zip CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a1963ba643eba7031835cfd0e704294420d5f446cb4e64a9b2ce6564a11eafc4
3
- size 5701
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f7b9cb283b2985efeaf612ba2cf7f116e0a9a992f5a859c9f6e231440e35d757
3
+ size 35875