File size: 687 Bytes
7f51798
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# 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 *