summaryrefslogtreecommitdiff
path: root/training
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-25 23:58:11 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-06-25 23:59:07 +0200
commitbc8184728044ef823537ca2358cd1cd9c22e1b9e (patch)
tree9e9a902fee0941f888be5b9a2597a56be5bedf77 /training
parentbc937ad0b430e0d61b3d147a7408ee946c425e0c (diff)
Update artifact name for checkpoints
Diffstat (limited to 'training')
-rw-r--r--training/callbacks/wandb_callbacks.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/training/callbacks/wandb_callbacks.py b/training/callbacks/wandb_callbacks.py
index 274a8b7..9e1d4d4 100644
--- a/training/callbacks/wandb_callbacks.py
+++ b/training/callbacks/wandb_callbacks.py
@@ -70,7 +70,7 @@ class UploadCheckpointsAsArtifact(Callback):
"""Uploads model checkpoint to W&B."""
logger = get_wandb_logger(trainer)
experiment = logger.experiment
- ckpts = wandb.Artifact("experiment-ckpts", type="checkpoints")
+ ckpts = wandb.Artifact("checkpoints", type="checkpoints")
if self.upload_best_only:
ckpts.add_file(trainer.checkpoint_callback.best_model_path)