Wuvin's picture
init
37aeb5b
raw
history blame
274 Bytes
__modules__ = {}
def register(name):
def decorator(cls):
__modules__[name] = cls
return cls
return decorator
def find(name):
return __modules__[name]
from custum_3d_diffusion.trainings import base, image2mvimage_trainer, image2image_trainer