diff options
| author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-25 23:57:04 +0200 |
|---|---|---|
| committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-06-25 23:57:04 +0200 |
| commit | 9e0f169f9188856e0bd78f3b224b7874cdb39651 (patch) | |
| tree | f81412910a826283fc361d2dc40a3027a823ea95 /training/callbacks | |
| parent | a30acba378f6f8fb86bb1a6096b9eb0ae7a68449 (diff) | |
Fix config path in wandb callback
Diffstat (limited to 'training/callbacks')
| -rw-r--r-- | training/callbacks/wandb_callbacks.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py index f98cfa5..9579f7e 100644 --- a/training/callbacks/wandb_callbacks.py +++ b/training/callbacks/wandb_callbacks.py @@ -42,7 +42,7 @@ class UploadConfigAsArtifact(Callback): """Upload all *.py files to W&B as an artifact, at the beginning of the run.""" def __init__(self) -> None: - self.config_dir = Path(".hydra/") + self.config_dir = Path(".") @rank_zero_only def on_train_start(self, trainer: Trainer, pl_module: LightningModule) -> None: |