cfb7702
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
models = {}
def register(name):
def decorator(cls):
models[name] = cls
return cls
return decorator
def make(name, config):
model = models[name](config)
return model
from . import nerf, neus, geometry, texture
|