562c323823c1cde522f57b0ebb8496f561e949c7,examples/transformer/hyperparams.py,,,#,14

Before Change

    "seed": 123,
    "shuffle": False,
    "source_dataset" : {
        "files": [os.path.join(args.data_dir, args.dev_src)],
        "vocab_file": args.vocab_file,
    },
    "target_dataset": {
        "files": [os.path.join(args.data_dir, args.dev_tgt)],
        "vocab_share": True,
    },
    "batch_size": args.test_batch_size,
    "allow_smaller_final_batch": True,
}
test_dataset_hparams = {
    "num_epochs": 1,
    "seed": 123,
    "shuffle": False,
    "source_dataset": {
        "files": [os.path.join(args.data_dir, args.test_src)],
        "vocab_file": args.vocab_file,
    },
    "target_dataset": {
        "files": [os.path.join(args.data_dir, args.test_tgt)],
        "vocab_share":True,
    },
    "batch_size": args.test_batch_size,

After Change

)
batching_scheme["boundaries"] = [b + 1 for b in batching_scheme["boundaries"]]
print("train_src:{}".format(args.train_src))
print("dev src:{}".format(args.dev_src))
train_dataset_hparams = {
    "num_epochs": args.num_epochs,
    //"num_epochs": args.max_train_epoch,

Instances