import paddlepalm as palm

if __name__ == '__main__':
    controller = palm.Controller('config.yaml', task_dir='tasks')
    controller.load_pretrain('../../pretrain/ernie-en-uncased-large/params')
    controller.train()

    controller = palm.Controller(config='config.yaml', task_dir='tasks', for_train=False)
    controller.pred('atis_slot', inference_model_dir='output_model/fourthrun/atis_slot/infer_model')