from .apps.DHU.train import train as train_DHU def main(*args,config=None): mode = config['mode'] equp_type = config['equp_type'] if mode == 'train' and equp_type == 'DHU': train_DHU(*args,config=config) else: raise NotImplementedError