Browse Source

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

NH-Rep
mckay 5 months ago
parent
commit
be50ef39fc
  1. 3
      code/conversion/run.py

3
code/conversion/run.py

@ -1,6 +1,7 @@
import os import os
import sys import sys
import time import time
project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir)) project_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir))
sys.path.append(project_dir) sys.path.append(project_dir)
os.chdir(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 utils.plots import plot_surface, plot_cuts_axis
from torch.utils.tensorboard import SummaryWriter from torch.utils.tensorboard import SummaryWriter
import torch.nn.functional as F import torch.nn.functional as F
from utils.logger import logger
logger.info(f"project_dir: {project_dir}")
class ReconstructionRunner: class ReconstructionRunner:
def run_nhrepnet_training(self): def run_nhrepnet_training(self):

Loading…
Cancel
Save