1db3e4b5d647adef9811fa74ab8c6bf3638ade62,onmt/Translator.py,Translator,__init__,#Translator#Any#,7

Before Change

        self.tt = torch.cuda if opt.cuda else torch

        checkpoint = torch.load(opt.model)
        self.model = checkpoint["model"]

        self.model.eval()

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

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

After Change

            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()

Instances