Merge branch 'fix-autotune'

thatguy11325 · Nov 27, 2024 · 91a6573 · 91a6573
2 parents 70be9fc + bc20bfc
commit 91a6573
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pokemonred_puffer/train.py b/pokemonred_puffer/train.py
@@ -250,17 +250,19 @@ def autotune(
             "-r",
             help="Reward module to use in rewards",
         ),
-    ] = "baseline.BaselineRewardEnv",
+    ] = "baseline.ObjectRewardRequiredEventsMapIds",
     wrappers_name: Annotated[
         str,
         typer.Option(
             "--wrappers-name",
             "-w",
             help="Wrappers to use _in order of instantion_",
         ),
-    ] = "baseline",
+    ] = "empty",
     rom_path: Path = "red.gb",
 ):
+    config = load_from_config(config, False)
+    config.vectorization = "multiprocessing"
     config, env_creator = setup(
         config=config,
         debug=False,