Replace HalfCheetah-v5 policy bundle

README.md

@@ -80,7 +80,7 @@ Notebook version: [examples/hub_quickstart.ipynb](https://github.com/ccnets-team
 | Env | Bundle | Ctx | Return | Norm. | Medium Ref. |
 |---|---|---:|---:|---:|---:|
 | `Ant-v5` | `ant-v5` | 32 | 3339.51±1115.40 | 50.56±16.54 | 86.54 |
-| `HalfCheetah-v5` | `halfcheetah-v5` | 32 | 4877.39±1899.50 | 31.12±11.51 | 74.83 |
+| `HalfCheetah-v5` | `halfcheetah-v5` | 32 | 5989.04±1902.22 | 37.86±11.53 | 74.83 |
 | `Hopper-v5` | `hopper-v5` | 32 | 2836.28±987.67 | 73.40±25.72 | 72.91 |
 | `Walker2d-v5` | `walker2d-v5` | 32 | 3883.30±684.09 | 56.69±9.99 | 83.26 |
 | `Humanoid-v5` | `humanoid-v5` | 32 | 6089.64±2512.73 | 70.41±29.58 | 81.30 |

halfcheetah-v5/config.json

@@ -1,15 +1,14 @@
 {
   "bundle_format_version": 1,
-  "package_version": "0.2.1",
+  "package_version": "0.2.2",
   "model_config": {
     "network_name": "Llama",
     "num_layers": 4,
-    "d_model": 192,
-    "num_heads": 12,
-    "dropout": 0.0,
-    "init_log_std": -1.0,
+    "d_model": 256,
+    "num_heads": 8,
+    "dropout": 0.05,
     "rope_theta": 1000.0,
-    "intermediate_size": 768,
+    "intermediate_size": 1024,
     "max_position_embeddings": 256,
     "context_length": 32
   },

halfcheetah-v5/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cef5f897bedf5326e1632a84b844eb1ae978008fa50783358d28033c1e0ad888
-size 9479980
+oid sha256:16840920704099d661ee4bf1a498cdae57d5f02efd8538633ad394b21c64f769
+size 16832380