update

triplaneturbo_executable/utils/saving.py

@@ -13,7 +13,7 @@ import wandb
 from matplotlib import cm
 from matplotlib.colors import LinearSegmentedColormap
 from PIL import Image, ImageDraw
-from pytorch_lightning.loggers import WandbLogger
+# from pytorch_lightning.loggers import WandbLogger
 
 from ..utils.mesh import Mesh
 
@@ -26,7 +26,7 @@ import threading
 
 class SaverMixin:
     _save_dir: Optional[str] = None
-    _wandb_logger: Optional[WandbLogger] = None
+    # _wandb_logger: Optional[WandbLogger] = None
 
     def set_save_dir(self, save_dir: str):
         self._save_dir = save_dir
@@ -58,17 +58,17 @@ class SaverMixin:
         os.makedirs(os.path.dirname(save_path), exist_ok=True)
         return save_path
 
-    def create_loggers(self, cfg_loggers: DictConfig) -> None:
-        if "wandb" in cfg_loggers.keys() and cfg_loggers.wandb.enable:
-            self._wandb_logger = WandbLogger(
-                project=cfg_loggers.wandb.project, name=cfg_loggers.wandb.name
-            )
+    # def create_loggers(self, cfg_loggers: DictConfig) -> None:
+    #     if "wandb" in cfg_loggers.keys() and cfg_loggers.wandb.enable:
+    #         self._wandb_logger = WandbLogger(
+    #             project=cfg_loggers.wandb.project, name=cfg_loggers.wandb.name
+    #         )
 
-    def get_loggers(self) -> List:
-        if self._wandb_logger:
-            return [self._wandb_logger]
-        else:
-            return []
+    # def get_loggers(self) -> List:
+    #     if self._wandb_logger:
+    #         return [self._wandb_logger]
+    #     else:
+    #         return []
 
     DEFAULT_RGB_KWARGS = {"data_format": "HWC", "data_range": (0, 1)}
     DEFAULT_UV_KWARGS = {