From bc8184728044ef823537ca2358cd1cd9c22e1b9e Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sat, 25 Jun 2022 23:58:11 +0200 Subject: Update artifact name for checkpoints --- training/callbacks/wandb_callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'training/callbacks') 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) -- cgit v1.2.3-70-g09d2