fd4b66763f91bf6e940184fb01a264c941a2e90e,pythainlp/transliterate/thai2rom.py,ThaiTransliterator,__init__,#ThaiTransliterator#,15

Before Change

            self.__filemodel = get_corpus_path("thai2rom-pytorch")
        loader = torch.load(self.__filemodel, map_location=device)
        self._n_h = 64  // hidden dimensions for encoder
        self._n_s = 64  // hidden dimensions for decoder
        self._emb_dim = 64  // character embedding size
        self._maxlength = 100
        self._char_to_ix = loader["char_to_ix"]
        self._ix_to_char = loader["ix_to_char"]
        self._target_char_to_ix = loader["target_char_to_ix"]
        self._ix_to_target_char = loader["ix_to_target_char"]
        // encoder/ decoder
        // Restore the model and construct the encoder and decoder.
        self._encoder = Encoder(len(self._char_to_ix),
                                self._n_h, self._emb_dim).to(device)
        self._encoder.load_state_dict(loader["encoder_state_dict"])
        self._decoder = OneStepDecoder(
            len(self._target_char_to_ix), self._n_s, self._emb_dim).to(device)
        self._decoder.load_state_dict(loader["decoder_state_dict"])

    def _prepare_sequence_in(self, input_text):
        

After Change

        loader = torch.load(self.__filemodel, map_location=device)

        INPUT_DIM, ENC_EMB_DIM, ENC_HID_DIM, ENC_DROPOUT = loader[
                                                            "encoder_params"
                                                           ]
        OUTPUT_DIM, DEC_EMB_DIM, DEC_HID_DIM, DEC_DROPOUT = loader[
                                                             "decoder_params"
                                                            ]

        self._maxlength = 100

        self._char_to_ix = loader["char_to_ix"]
        self._ix_to_char = loader["ix_to_char"]
        self._target_char_to_ix = loader["target_char_to_ix"]
        self._ix_to_target_char = loader["ix_to_target_char"]

        // encoder/ decoder
        // Restore the model and construct the encoder and decoder.
        self._encoder = Encoder(
            INPUT_DIM,
            ENC_EMB_DIM,
            ENC_HID_DIM,
            ENC_DROPOUT
        )

        self._decoder = AttentionDecoder(
            OUTPUT_DIM,
            DEC_EMB_DIM,

Instances