from util.registry import Registry MODEL = Registry("model")