Added parallel training to alpha zero.

Skirlax · Aug 13, 2024 · 1547f64 · 1547f64
1 parent 7b5a898
commit 1547f64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mu_alpha_zero/AlphaZero/utils.py b/mu_alpha_zero/AlphaZero/utils.py
@@ -145,7 +145,7 @@ def get_function_from_value(value, trial: optuna.Trial):
         if type(value[1]) == list:
             return trial.suggest_categorical(value[0], value[1])
 
-    def objective(trial: optuna.Trial):
+    def objective(trial: optuna.Trial,az,config):
         for value in target_values:
             setattr(config, value[0], get_function_from_value(value, trial))