vec

Learning-and-Intelligent-Systems · Sep 15, 2024 · c9ad1e8 · c9ad1e8
1 parent 13cbc36
commit c9ad1e8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/predicators/ml_models.py b/predicators/ml_models.py
@@ -1737,7 +1737,7 @@ def _vectorize_state(self, state: State) -> Array:
                 padded_e = np.pad(e, (0, pad_length - len(e)), 'constant')
                 return padded_e
             elif CFG.env == "coffee":
-                pad_length = 20
+                pad_length = 40
                 padded_e = np.pad(e, (0, pad_length - len(e)), 'constant')
                 return padded_e
             pad_length = 300
@@ -2086,6 +2086,9 @@ def train_q_function(self) -> None:
             X_size = 5 + 7 + self._max_num_params
         elif CFG.env in {"coffee", "coffeelids"} and CFG.use_obj_centric:
             X_size = 5 + 7 + self._max_num_params -1
+        else:
+            X_size = 48
+
 
         Y_size = 1
         doorknob_values = []