From 4795a9965b455503d7475bcbc2f2fcba91f14757 Mon Sep 17 00:00:00 2001 From: Alexandra Udaltsova <43303448+AUdaltsova@users.noreply.github.com> Date: Fri, 20 Sep 2024 16:36:50 +0100 Subject: [PATCH] Fix wandb artifact saving in base_model.py --- pvnet/models/base_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pvnet/models/base_model.py b/pvnet/models/base_model.py index e6e0e907..8665bf3c 100644 --- a/pvnet/models/base_model.py +++ b/pvnet/models/base_model.py @@ -735,7 +735,7 @@ def on_validation_epoch_end(self): # make and log wand artifact validation_artifact = wandb.Artifact( - f"validation_results_epoch={self.current_epoch}", type="dataset" + f"validation_results_epoch_{self.current_epoch}", type="dataset" ) validation_artifact.add_file(filename) wandb.log_artifact(validation_artifact)