15 lines
274 B
Python
15 lines
274 B
Python
__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
|