Python run Examples

Example #1

File: test_transformer.py Project: Aaronkiim/NJUNMT-pytorch

def test_transformer_train(test_dir):
    from src.bin import train

    config_path = "./unittests/configs/test_transformer.yaml"
    model_name = test_utils.get_model_name(config_path)

    saveto = os.path.join(test_dir, "save")
    log_path = os.path.join(test_dir, "log")
    valid_path = os.path.join(test_dir, "valid")

    train.run(model_name=model_name,
              config_path=config_path,
              saveto=saveto,
              log_path=log_path,
              valid_path=valid_path,
              debug=True)

Example #2

File: test_ocd_loss.py Project: whr94621/ODC-NMT

def test_transformer_train(test_dir, use_gpu=False, multi_gpu=False):
    from src.bin import train

    config_path = "./unittests/configs/test_ocd_loss.yaml"
    model_name = test_utils.get_model_name(config_path)

    saveto = os.path.join(test_dir, "save")
    log_path = os.path.join(test_dir, "log")
    valid_path = os.path.join(test_dir, "valid")

    train.run(model_name=model_name,
              config_path=config_path,
              saveto=saveto,
              log_path=log_path,
              valid_path=valid_path,
              debug=True,
              use_gpu=use_gpu,
              multi_gpu=multi_gpu,
              task="ocd_old")

Example #3

File: train.py Project: howardchenhd/Paraphrase_Copy

from src.bin import train

if __name__ == '__main__':

    train.run()