Skip to content

Commit 16b6b3f

Browse files
Fix: Remove duplicate W&B initialization in offline mode (#3886)
1 parent 11e97ab commit 16b6b3f

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

src/accelerate/tracking.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -340,16 +340,7 @@ def store_init_configuration(self, values: dict):
340340
"""
341341
import wandb
342342

343-
if os.environ.get("WANDB_MODE") == "offline":
344-
# In offline mode, restart wandb with config included
345-
if hasattr(self, "run") and self.run:
346-
self.run.finish()
347-
348-
init_kwargs = self.init_kwargs.copy()
349-
init_kwargs["config"] = values
350-
self.run = wandb.init(project=self.run_name, **init_kwargs)
351-
else:
352-
wandb.config.update(values, allow_val_change=True)
343+
wandb.config.update(values, allow_val_change=True)
353344
logger.debug("Stored initial configuration hyperparameters to WandB")
354345

355346
@on_main_process

0 commit comments

Comments
 (0)