Merge pull request #254 from openclimatefix/wandb-patch

Fix wandb artifact saving in base_model.py
openclimatefix · Sep 23, 2024 · ad30b23 · ad30b23
2 parents 76651e9 + 4795a99
commit ad30b23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)