feat: update SheepRLv0.4.8

diambra · Nov 29, 2023 · 8b651f0 · 8b651f0
1 parent f9c86d2
commit 8b651f0
Showing 1 changed file with 19 additions and 19 deletions.
diff --git a/sheeprl/configs/exp/custom_exp.yaml b/sheeprl/configs/exp/custom_exp.yaml
@@ -6,10 +6,6 @@ defaults:
   - override /env: custom_env
   - _self_
 
-# Experiment
-total_steps: 1024
-per_rank_batch_size: 16
-
 # Buffer
 buffer:
   share_data: False
@@ -18,18 +14,22 @@ buffer:
 checkpoint:
   save_last: True
 
-cnn_keys:
-  encoder: [frame]
-mlp_keys:
-  encoder:
-    - own_character
-    - own_health
-    - own_side
-    - own_wins
-    - opp_character
-    - opp_health
-    - opp_side
-    - opp_wins
-    - stage
-    - timer
-    - action
+# Experiment
+algo:
+  total_steps: 1024
+  per_rank_batch_size: 16
+  cnn_keys:
+    encoder: [frame]
+  mlp_keys:
+    encoder:
+      - own_character
+      - own_health
+      - own_side
+      - own_wins
+      - opp_character
+      - opp_health
+      - opp_side
+      - opp_wins
+      - stage
+      - timer
+      - action