Python main.train方法代碼示例

本文整理匯總了Python中main.train方法的典型用法代碼示例。如果您正苦於以下問題：Python main.train方法的具體用法？Python main.train怎麽用？Python main.train使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類main的用法示例。

在下文中一共展示了main.train方法的12個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。

示例1: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    if config.mode == "train":
        train(config)
    elif config.mode == "prepro":
        prepro(config)
    elif config.mode == "debug":
        config.num_steps = 2
        config.val_num_batches = 1
        config.checkpoint = 1
        config.period = 1
        train(config)
    elif config.mode == "test":
        test(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:HKUST-KnowComp，項目名稱:R-Net，代碼行數:19，代碼來源:config.py

示例2: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    if config.mode == "train":
        train(config)
    elif config.mode == "prepro":
        prepro(config)
    elif config.mode == "debug":
        config.num_steps = 2
        config.val_num_batches = 1
        config.checkpoint = 1
        config.period = 1
        train(config)
    elif config.mode == "test":
        test(config)
    else:
        print("Unknown mode, you must choose mode from [train/prepro/debug/test]")
        exit(0)

開發者ID:l11x0m7，項目名稱:Question_Answering_Models，代碼行數:19，代碼來源:config.py

示例3: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    if config.mode == "train":
        train(config)
    elif config.mode == "prepro":
        prepro(config)
    elif config.mode == "debug":
        config.num_steps = 2
        config.val_num_batches = 1
        config.checkpoint = 1
        config.period = 1
        train(config)
    elif config.mode == "test":
        test(config)
    elif config.mode == "demo":
        demo(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:NLPLearn，項目名稱:QANet，代碼行數:21，代碼來源:config.py

示例4: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    os.environ["CUDA_VISIBLE_DEVICES"] = config.gpu  # 選擇一塊gpu
    if config.mode == "train":
        train(config)
    elif config.mode == "prepro":
        data_process.prepro(config)
    elif config.mode == "debug":
        config.num_steps = 2
        config.val_num_batches = 1
        config.checkpoint = 1
        config.period = 1
        train(config)
    elif config.mode == "test":
        test(config)
    elif config.mode == "examine":
        examine_dev(config)
    elif config.mode == "save_dev":
        save_dev(config)
    elif config.mode == "save_test":
        save_test(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:yuhaitao1994，項目名稱:AIchallenger2018_MachineReadingComprehension，代碼行數:26，代碼來源:config.py

示例5: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    os.environ["CUDA_VISIBLE_DEVICES"] = config.gpu  # 選擇一塊gpu
    if config.mode == "train":
        train(config)
    elif config.mode == "prepro":
        data_process_addAnswer.prepro(config)
    elif config.mode == "test":
        test(config)
    elif config.mode == "examine":
        examine_dev(config)
    elif config.mode == "save_dev":
        save_dev(config)
    elif config.mode == "save_test":
        save_test(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:yuhaitao1994，項目名稱:AIchallenger2018_MachineReadingComprehension，代碼行數:20，代碼來源:config.py

示例6: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    if config.mode == "get_vocab":
        get_vocab(config)
    elif config.mode == "prepare":
        prepare(config)
    elif config.mode == "train":
        train(config)
    elif config.mode == "train_rl":
        train_rl(config)
    elif config.mode == "train_qpp":
        train_qpp(config)
    elif config.mode == "train_qap":
        train_qap(config)
    elif config.mode == "train_qqp_qap":
        train_qqp_qap(config)
    elif config.mode == "test":
        test(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:ZhangShiyue，項目名稱:QGforQA，代碼行數:23，代碼來源:config.py

示例7: run

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def run(_):
    if config.mode == "train":
        train(config,device)
    elif config.mode == "preprocess":
        preproc(config)

    elif config.mode == "debug":
        config.epochs = 1
        config.batch_size = 3
        config.val_batch_size = 20
        config.checkpoint = 1
        config.period = 1
        train(config,device)
    elif config.mode == "test":
        pass
    elif config.mode == "dev":
        dev(config,device)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:InitialBug，項目名稱:BiSET，代碼行數:22，代碼來源:config.py

示例8: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS

    if config.mode == 'prepro':
        prepro(config)
    elif config.mode == 'train':
        train(config)
    elif config.mode == 'test':
        test(config)
    else:
        raise ValueError('The mode must choose from [prepro/train/test]!')

開發者ID:l11x0m7，項目名稱:Question_Answering_Models，代碼行數:13，代碼來源:config.py

示例9: train

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def train():
    model_num, model_name = get_latest_model()
    print("Training on gathered game data, initializing from {}".format(model_name))
    new_model_name = shipname.generate(model_num + 1)
    print("New model will be {}".format(new_model_name))
    load_file = os.path.join(MODELS_DIR, model_name)
    save_file = os.path.join(MODELS_DIR, new_model_name)
    #try:
    main.train(ESTIMATOR_WORKING_DIR, TRAINING_CHUNK_DIR, save_file,
               generation_num=model_num + 1)
    #except:
    #    print("Got an error training, muddling on...")
    #    logging.exception("Train error")
    return new_model_name

開發者ID:mlperf，項目名稱:training_results_v0.5，代碼行數:16，代碼來源:loop_train_eval.py

示例10: rl_loop_train

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def rl_loop_train():
    """Run the reinforcement learning loop

    This tries to create a realistic way to run the reinforcement learning with
    all default parameters.
    """
    qmeas.stop_time('selfplay_wait')
    print("Gathering game output...")
    gather()

    print("Training on gathered game data...")
    _, model_name = get_latest_model()
    new_model = train()

開發者ID:mlperf，項目名稱:training_results_v0.5，代碼行數:15，代碼來源:loop_train_eval.py

示例11: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    if config.mode == "prepare":
        prepare(config)
    elif config.mode == "test":
        test(config)
    elif config.mode == "train":
        train(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:ZhangShiyue，項目名稱:QGforQA，代碼行數:13，代碼來源:config.py

示例12: main

# 需要導入模塊: import main [as 別名]
# 或者: from main import train [as 別名]
def main(_):
    config = flags.FLAGS
    if config.mode == "prepare":
        prepare(config)
    elif config.mode == "train":
        train(config)
    elif config.mode == "test":
        test(config)
    else:
        print("Unknown mode")
        exit(0)

開發者ID:ZhangShiyue，項目名稱:QGforQA，代碼行數:13，代碼來源:config.py

注：本文中的main.train方法示例由純淨天空整理自Github/MSDocs等開源代碼及文檔管理平台，相關代碼片段篩選自各路編程大神貢獻的開源項目，源碼版權歸原作者所有，傳播和使用請參考對應項目的License；未經允許，請勿轉載。