diff --git a/open_diloco/utils.py b/open_diloco/utils.py index 9940747..d036519 100644 --- a/open_diloco/utils.py +++ b/open_diloco/utils.py @@ -177,7 +177,7 @@ def finish(self): ... class WandbLogger: def __init__(self, project, config): - wandb.init(project=project, config=config) + wandb.init(project=project, config=config, resume="auto") # make wandb reuse the same run id if possible def log(self, metrics: dict[str, Any]): wandb.log(metrics)