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