kissin42 commited on
Commit
ef26218
·
verified ·
1 Parent(s): 65c22ce

Replace Hopper-v5 with stable bundle 24

Browse files

Causal GPT-RL bundle format 2, package 0.14.0. Gymnasium 1.2.3, MuJoCo 3.2.3, seeds 0..49, max_steps=1000, BOS discard. Main score uses kv_cache_max_len=32; the same protocol was used for kv=16 and kv=64.

README.md CHANGED
@@ -81,7 +81,7 @@ Notebook version: [examples/hub_quickstart.ipynb](https://github.com/ccnets-team
81
  |---|---|---:|---:|---:|---:|---:|
82
  | `Ant-v5` | `ant-v5` | 32 | 5126.03±1568.43 | 77.05±23.26 | 59.99 ✓ | 86.54 ✗ |
83
  | `HalfCheetah-v5` | `halfcheetah-v5` | 32 | 6816.48±3135.53 | 42.87±19.01 | 43.54 ✗ | 74.83 ✗ |
84
- | `Hopper-v5` | `hopper-v5` | 32 | 2713.66±1075.57 | 70.21±28.01 | 42.65 ✓ | 72.91 |
85
  | `Walker2d-v5` | `walker2d-v5` | 32 | 3899.88±706.57 | 56.93±10.32 | 59.51 ✗ | 83.26 ✗ |
86
  | `Humanoid-v5` | `humanoid-v5` | 32 | 7892.65±1018.11 | 91.63±11.99 | 63.29 ✓ | 81.30 ✓ |
87
 
@@ -111,7 +111,7 @@ seeds `0..49`, `max_steps=1000`):
111
  |---|---:|---:|---:|---|
112
  | `Ant-v5` | 5109.97±1711.18 | 5126.03±1568.43 | **5256.33±1429.58** | **2× highest mean (≈)** |
113
  | `HalfCheetah-v5` | 6793.17±2939.17 | 6816.48±3135.53 | 6468.21±3234.51 | ≈ flat |
114
- | `Hopper-v5` | 3361.71±103.69 | 2713.66±1075.57 | 992.92±445.63 | shorter ↑, collapses |
115
  | `Walker2d-v5` | 3950.09±459.01 | 3899.88±706.57 | 3842.19±718.60 | ≈ flat |
116
  | `Humanoid-v5` | 7431.52±2024.95 | 7892.65±1018.11 | 8040.41±38.02 | longer ↑, steadiest |
117
 
 
81
  |---|---|---:|---:|---:|---:|---:|
82
  | `Ant-v5` | `ant-v5` | 32 | 5126.03±1568.43 | 77.05±23.26 | 59.99 ✓ | 86.54 ✗ |
83
  | `HalfCheetah-v5` | `halfcheetah-v5` | 32 | 6816.48±3135.53 | 42.87±19.01 | 43.54 ✗ | 74.83 ✗ |
84
+ | `Hopper-v5` | `hopper-v5` | 32 | 3199.65±21.74 | 82.87±0.57 | 42.65 ✓ | 72.91 |
85
  | `Walker2d-v5` | `walker2d-v5` | 32 | 3899.88±706.57 | 56.93±10.32 | 59.51 ✗ | 83.26 ✗ |
86
  | `Humanoid-v5` | `humanoid-v5` | 32 | 7892.65±1018.11 | 91.63±11.99 | 63.29 ✓ | 81.30 ✓ |
87
 
 
111
  |---|---:|---:|---:|---|
112
  | `Ant-v5` | 5109.97±1711.18 | 5126.03±1568.43 | **5256.33±1429.58** | **2× highest mean (≈)** |
113
  | `HalfCheetah-v5` | 6793.17±2939.17 | 6816.48±3135.53 | 6468.21±3234.51 | ≈ flat |
114
+ | `Hopper-v5` | 3189.53±22.58 | **3199.65±21.74** | 3190.09±23.75 | **≈ flat and stable** |
115
  | `Walker2d-v5` | 3950.09±459.01 | 3899.88±706.57 | 3842.19±718.60 | ≈ flat |
116
  | `Humanoid-v5` | 7431.52±2024.95 | 7892.65±1018.11 | 8040.41±38.02 | longer ↑, steadiest |
117
 
hopper-v5/config.json CHANGED
@@ -1,14 +1,16 @@
1
  {
2
  "bundle_format_version": 2,
3
- "package_version": "0.2.2",
4
  "model_config": {
5
  "network_name": "Llama",
6
  "num_layers": 4,
7
- "d_model": 192,
8
- "num_heads": 6,
9
  "dropout": 0.05,
 
 
10
  "rope_theta": 1000.0,
11
- "intermediate_size": 768,
12
  "max_position_embeddings": 256,
13
  "context_length": 32
14
  },
@@ -65,9 +67,63 @@
65
  }
66
  ],
67
  "context_length": 32,
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
68
  "state_normalization": {
69
  "embedded": true,
70
  "legacy_sidecar": false
71
  },
72
- "env_id": "Hopper-v5"
 
 
 
73
  }
 
1
  {
2
  "bundle_format_version": 2,
3
+ "package_version": "0.14.0",
4
  "model_config": {
5
  "network_name": "Llama",
6
  "num_layers": 4,
7
+ "d_model": 256,
8
+ "num_heads": 8,
9
  "dropout": 0.05,
10
+ "use_eos": true,
11
+ "use_bos_action_gate": true,
12
  "rope_theta": 1000.0,
13
+ "intermediate_size": 1024,
14
  "max_position_embeddings": 256,
15
  "context_length": 32
16
  },
 
67
  }
68
  ],
69
  "context_length": 32,
70
+ "requires_capabilities": [],
71
+ "state_container": {
72
+ "type": "Box",
73
+ "shape": [
74
+ 11
75
+ ],
76
+ "dtype": "float64",
77
+ "low": [
78
+ "-inf",
79
+ "-inf",
80
+ "-inf",
81
+ "-inf",
82
+ "-inf",
83
+ "-inf",
84
+ "-inf",
85
+ "-inf",
86
+ "-inf",
87
+ "-inf",
88
+ "-inf"
89
+ ],
90
+ "high": [
91
+ "inf",
92
+ "inf",
93
+ "inf",
94
+ "inf",
95
+ "inf",
96
+ "inf",
97
+ "inf",
98
+ "inf",
99
+ "inf",
100
+ "inf",
101
+ "inf"
102
+ ]
103
+ },
104
+ "action_container": {
105
+ "type": "Box",
106
+ "shape": [
107
+ 3
108
+ ],
109
+ "dtype": "float32",
110
+ "low": [
111
+ -1.0,
112
+ -1.0,
113
+ -1.0
114
+ ],
115
+ "high": [
116
+ 1.0,
117
+ 1.0,
118
+ 1.0
119
+ ]
120
+ },
121
  "state_normalization": {
122
  "embedded": true,
123
  "legacy_sidecar": false
124
  },
125
+ "env_id": "Hopper-v5",
126
+ "serving": {
127
+ "bos_cache_mode": "discard"
128
+ }
129
  }
hopper-v5/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:72db9780f52bfd27deb03f069b018ef61af0e8dcaef8a92c9c48c72ef686d700
3
- size 9468696
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e8703976cec44a1b24aa3938a4f48ce2a5f1463c39a2e688946146f52b3a8bb5
3
+ size 16818584