fix again

thatguy11325 · Aug 12, 2024 · 89b2582 · 89b2582
1 parent 1c92052
commit 89b2582
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pokemonred_puffer/policies/multi_convolutional.py b/pokemonred_puffer/policies/multi_convolutional.py
@@ -233,7 +233,13 @@ def encode_observations(self, observations):
             dim=-1,
         )
         if self.skip_safari_zone:
-            cat_obs = torch.cat((cat_obs, observations["safari_steps"].float()), dim=-1)
+            cat_obs = torch.cat(
+                (
+                    cat_obs,
+                    observations["safari_steps"].float() / 502.0,
+                ),
+                dim=-1,
+            )
         if self.use_global_map:
             cat_obs = torch.cat(
                 (