Python update_configの例

コード例 #1

ファイル: main_arcii.py プロジェクト: Junpliu/QAMODEL

def main(unused):

    misc_utils.update_config(FLAGS)

    model_creator = ARCII.ARCII

    # train
    if FLAGS.train:
        logger.info("TRAIN")
        train.train(FLAGS, model_creator)

    # evaluate
    if FLAGS.test:
        logger.info("TEST")
        train.test(FLAGS, model_creator)

    # inference
    if FLAGS.infer:
        logger.info("INFER")
        inference.inference(FLAGS, model_creator)

コード例 #2

def main(unused):

    misc_utils.update_config(FLAGS)

    model_creator = SALSTM.SALSTM

    # train
    if FLAGS.train:
        logger.info("TRAIN")
        train.train(FLAGS, model_creator)
        # TODO: export model for serving
        train.export_model(FLAGS, model_creator)

    # evaluate
    if FLAGS.test:
        logger.info("TEST")
        train.test(FLAGS, model_creator)

    # inference
    if FLAGS.infer:
        logger.info("INFER")
        inference.inference(FLAGS, model_creator)