From be50ef39fc901d9060bd7defbcfa5f4404ff63dd Mon Sep 17 00:00:00 2001
From: mckay <wchpub@163.com>
Date: Sun, 5 Jan 2025 14:43:14 +0800
Subject: [PATCH] Enhance logging in run.py by adding project directory info
 and initializing logger. This improves traceability during execution.

---
 code/conversion/run.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/code/conversion/run.py b/code/conversion/run.py
index 1a37737..6dbad5b 100644
--- a/code/conversion/run.py
+++ b/code/conversion/run.py
@@ -1,6 +1,7 @@
 import os
 import sys
 import time
+
 project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
 sys.path.append(project_dir)
 os.chdir(project_dir)
@@ -41,6 +42,8 @@ from scipy.spatial import cKDTree
 from utils.plots import plot_surface, plot_cuts_axis
 from torch.utils.tensorboard import SummaryWriter
 import torch.nn.functional as F
+from utils.logger import logger
+logger.info(f"project_dir: {project_dir}")
 
 class ReconstructionRunner:
     def run_nhrepnet_training(self):