Browse Source

Enhance logging in run.py by adding project directory info and initializing logger. This improves traceability during execution.

main
mckay 2 months ago
parent
commit
be50ef39fc
  1. 3
      code/conversion/run.py

3
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):

Loading…
Cancel
Save