Skip to content

Commit 7e75f9f

Browse files
authored
Add support for model name appendix (#546)
1 parent 97938fb commit 7e75f9f

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/art/utils/benchmarking/log_constant_metrics_wandb.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ async def log_constant_metrics_wandb(
99
model: art.Model,
1010
num_steps: int,
1111
split_metrics: dict[str, dict[str, float]],
12+
model_name_appendix: str | None = None,
1213
) -> None:
1314
"""
1415
Log constant metrics to W&B as horizontal lines across all training steps.
@@ -31,7 +32,7 @@ async def log_constant_metrics_wandb(
3132
"""
3233
run = wandb.init(
3334
project=model.project,
34-
name=model.name,
35+
name=model.name + f" {model_name_appendix}" if model_name_appendix else "",
3536
reinit="create_new",
3637
)
3738

0 commit comments

Comments
 (0)