Add the combo

qiaoting159753 · Aug 6, 2024 · 36ec704 · 36ec704
1 parent ad2b72f
commit 36ec704
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/MBRL_steve_config.json b/MBRL_steve_config.json
@@ -0,0 +1,28 @@
+{
+    "G": 1, 
+    "G_model": 0.2, 
+
+    "buffer_size": 1000000,
+    "batch_size": 256,
+
+    "max_steps_exploration": 256,
+    "max_steps_training": 1000000,
+    "number_steps_per_train_policy": 1,
+
+    "min_noise": 0.0,
+    "noise_scale": 0.1,
+    "noise_decay": 1.0,
+
+    "algorithm": "STEVE_MEAN",
+    "actor_lr": 3e-4,
+    "critic_lr": 3e-4,
+    "alpha_lr": 3e-4,
+    "world_model_lr": 0.001,
+    "gamma": 0.99,
+    "tau": 0.005,
+
+    "horizon": 3,
+    "num_world_models": 5,
+    "num_reward_models": 5,
+    "num_critic_models": 4
+}