nsr/__init__.py

# triplane, tensorF etc.
from .train_util import TrainLoop3DRec, TrainLoop3DRecTrajVis
# from .train_util_cvD import TrainLoop3DcvD

# from .cvD.nvsD import TrainLoop3DcvD_nvsD
# from .cvD.nvsD_nosr import TrainLoop3DcvD_nvsD_noSR
# from .cvD.nvsD_canoD import TrainLoop3DcvD_nvsD_canoD, TrainLoop3DcvD_nvsD_canoD_eg3d
# from .cvD.nvsD_canoD_mask import TrainLoop3DcvD_nvsD_canoD_canomask
# from .cvD.canoD import TrainLoop3DcvD_canoD
# from .cvD.nvsD_canoD_multiview import TrainLoop3DcvD_nvsD_canoD_multiview

# * difffusion trainer
from .train_util_diffusion import TrainLoop3DDiffusion
from .train_util_diffusion_dit import TrainLoop3DDiffusionDiT

from .lsgm import *