File size: 435 Bytes
94e8ee8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
systems = {}
def register(name):
def decorator(cls):
systems[name] = cls
return cls
return decorator
def make(name, config, load_from_checkpoint=None):
if load_from_checkpoint is None:
system = systems[name](config)
else:
system = systems[name].load_from_checkpoint(load_from_checkpoint, strict=False, config=config)
return system
from . import neus, neus_ortho, neus_pinhole
|