988ee51b15466231e9770f46dcb4d221866322a7,OpenNMT/train.py,,main,#,263

Before Change

        print("Loading model from checkpoint at %s" % opt.train_from)
        model.load_state_dict(checkpoint["model"])
        generator.load_state_dict(checkpoint["generator"])
        optim = checkpoint["optim"]
        opt.start_epoch = checkpoint["epoch"] + 1

    if len(opt.gpus) >= 1:
        model.cuda()

After Change

    if opt.train_from:
        optim.last_ppl = checkpoint["last_ppl"]
        optim.optimizer.load_state_dict(checkpoint["optimizer"])

    nParams = sum([p.nelement() for p in model.parameters()])
    print("* number of parameters: %d" % nParams)

Instances