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