Remove wand version

openclimatefix · Dec 11, 2023 · e3db3ea · e3db3ea
1 parent cfe079b
commit e3db3ea
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/pvnet/training.py b/pvnet/training.py
@@ -103,10 +103,8 @@ def train(config: DictConfig) -> Optional[float]:
         for callback in callbacks:
             log.info(f"{callback}")
             if isinstance(callback, ModelCheckpoint):
-                print(callback)
-                print(wandb_logger)
                 callback.dirpath = "/".join(
-                    callback.dirpath.split("/")[:-1] + [wandb_logger.version]
+                    callback.dirpath.split("/")[:-1]
                 )
                 # Also save model config here - this makes for easy model push to huggingface
                 os.makedirs(callback.dirpath, exist_ok=True)