[update] 20260312

Author: Lmy0217

utils/wandb.py

@@ -0,0 +1,37 @@
+import os
+
+import wandb
+
+import configs
+import utils
+
+
+class Wandb(object):
+
+    def __init__(self, cfg, data_cfg, run):
+        self.cfg = cfg
+        self.data_cfg = data_cfg
+        self.run = run
+        self.path = utils.path.get_path(cfg, data_cfg, run)
+        self.project = os.path.split(self.path)[-1]
+
+        if configs.env.log.wandb:
+            self.wan = wandb.init(
+                entity="myluo-cair",
+                project=self.project,
+                config={
+                    'model': self.cfg.dict(),
+                    'dataset': self.data_cfg.dict(),
+                    'run': self.run.dict(),
+                },
+            )
+        else:
+            self.wan = None
+
+    def __del__(self):
+        if self.wan is not None:
+            self.wan.finish()
+
+    def info(self, msg):
+        if self.wan is not None:
+            self.wan.log(msg)