988ee51b15466231e9770f46dcb4d221866322a7,OpenNMT/onmt/Translator.py,Translator,__init__,#Translator#Any#,7

Before Change

        if opt.cuda:
            self.model.cuda()
        else:
            self.model.cpu()

        self.src_dict = checkpoint["dicts"]["src"]
        self.tgt_dict = checkpoint["dicts"]["tgt"]

After Change

        checkpoint = torch.load(opt.model)

        model_opt = checkpoint["opt"]
        self.src_dict = checkpoint["dicts"]["src"]
        self.tgt_dict = checkpoint["dicts"]["tgt"]

        encoder = onmt.Models.Encoder(model_opt, self.src_dict)
        decoder = onmt.Models.Decoder(model_opt, self.tgt_dict)
        model = onmt.Models.NMTModel(encoder, decoder)

        generator = nn.Sequential(
            nn.Linear(model_opt.rnn_size, self.tgt_dict.size()),
            nn.LogSoftmax())

        model.load_state_dict(checkpoint["model"])
        generator.load_state_dict(checkpoint["generator"])

        if opt.cuda:
            model.cuda()
            generator.cuda()
        else:
            model.cpu()
            generator.cpu()

Instances