Replace Hopper-v5 with stable bundle 24
Browse filesCausal 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 +2 -2
- hopper-v5/config.json +61 -5
- hopper-v5/model.safetensors +2 -2
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 |
|
| 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` |
|
| 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.
|
| 4 |
"model_config": {
|
| 5 |
"network_name": "Llama",
|
| 6 |
"num_layers": 4,
|
| 7 |
-
"d_model":
|
| 8 |
-
"num_heads":
|
| 9 |
"dropout": 0.05,
|
|
|
|
|
|
|
| 10 |
"rope_theta": 1000.0,
|
| 11 |
-
"intermediate_size":
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e8703976cec44a1b24aa3938a4f48ce2a5f1463c39a2e688946146f52b3a8bb5
|
| 3 |
+
size 16818584
|