b25cc3a7e6a560c1fd47f36a43296f62eaaace36,transformer/Translator.py,Translator,__init__,#Translator#,13

Before Change

    def __init__(self, opt):
        self.opt = opt
        self.device = torch.device("cuda" if opt.cuda else "cpu")

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

After Change

        self.register_buffer("init_seq", torch.LongTensor([[trg_bos_idx]]))
        self.register_buffer(
            "blank_seqs", 
            torch.full((beam_size, max_seq_len), trg_pad_idx, dtype=torch.long))
        self.blank_seqs[:, 0] = self.trg_bos_idx
        self.register_buffer(
            "len_map", 

Instances