301b0cbf78e3de174a50fff682e142c985b3ae88,OpenNMT/train.py,,main,#,225

Before Change

    print("Building model...")

    if "model" in checkpoint:
        model = checkpoint["model"]
    else:
        encoder = onmt.Models.Encoder(opt, dicts["src"])

After Change

            start_decay_at=opt.start_decay_at
        )
    else:
        checkpoint = torch.load(opt.train_from)
        model = checkpoint["model"]
        optim = checkpoint["optim"]
        opt.start_epoch = checkpoint["epoch"]

    if opt.cuda:
        model.cuda()

Instances