caf9f875d5e2abd044e7fc286f837ef40a484e82,train.py,,main,#,275

Before Change

    // Load train and validate data.
    print("Loading train and validate data from "%s"" % opt.data)
    train = torch.load(opt.data + ".train.pt")
    valid = torch.load(opt.data + ".valid.pt")
    print(" * number of training sentences: %d" % len(train))
    print(" * maximum batch size: %d" % opt.batch_size)

    // Load checkpoint if we resume from a previous training.
    if opt.train_from:

After Change

    // Load train and validate data.
    train_dataset = load_dataset("train")
    valid_dataset = load_dataset("valid")
    print(" * maximum batch size: %d" % opt.batch_size)

    // Load checkpoint if we resume from a previous training.
    if opt.train_from:

Instances