tracking number of gpu's used for benchmarking

womboai · Dec 10, 2024 · 9e6593a · 9e6593a
1 parent cea9d9d
commit 9e6593a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/validator/weight_setting/wandb_manager.py b/validator/weight_setting/wandb_manager.py
@@ -78,6 +78,7 @@ def send_metrics(
         data = {
             "scores": scores or contest_state.get_scores(contest_state.benchmarks),
             "ranks": ranks or contest_state.get_ranks(scores),
+            "num_gpus": len(self.config["benchmarker_api"]),
         } | contest_state.model_dump()
 
         self._run.log(data=data)