We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 11e97ab commit 16b6b3fCopy full SHA for 16b6b3f
src/accelerate/tracking.py
@@ -340,16 +340,7 @@ def store_init_configuration(self, values: dict):
340
"""
341
import wandb
342
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)
+ wandb.config.update(values, allow_val_change=True)
353
logger.debug("Stored initial configuration hyperparameters to WandB")
354
355
@on_main_process
0 commit comments