choose project name

benchmarl/conf/experiment/base_experiment.yaml

@@ -87,6 +87,8 @@ evaluation_deterministic_actions: True
 
 # List of loggers to use, options are: wandb, csv, tensorboard, mflow
 loggers: []
+# Wandb project name
+project_name: "benchmarl"
 # Create a json folder as part of the output in the format of marl-eval
 create_json: True
 

benchmarl/experiment/experiment.py

@@ -94,6 +94,7 @@ class ExperimentConfig:
     evaluation_deterministic_actions: bool = MISSING
 
     loggers: List[str] = MISSING
+    project_name: str = MISSING
     create_json: bool = MISSING
 
     save_folder: Optional[str] = MISSING
@@ -535,6 +536,7 @@ def _setup_name(self):
 
     def _setup_logger(self):
         self.logger = Logger(
+            project_name=self.config.project_name,
             experiment_name=self.name,
             folder_name=str(self.folder_name),
             experiment_config=self.config,

benchmarl/experiment/logger.py

@@ -33,6 +33,7 @@ def __init__(
         model_name: str,
         group_map: Dict[str, List[str]],
         seed: int,
+        project_name: str,
     ):
         self.experiment_config = experiment_config
         self.algorithm_name = algorithm_name
@@ -63,7 +64,7 @@ def __init__(
                     experiment_name=experiment_name,
                     wandb_kwargs={
                         "group": task_name,
-                        "project": "benchmarl",
+                        "project": project_name,
                         "id": experiment_name,
                     },
                 )