7f2f766be0b1b2d77d02621ec9f64ea9104fade0,parlai/agents/transformer/polyencoder.py,PolyencoderAgent,_set_text_vec,#PolyencoderAgent#,100

Before Change

         Add the start and end token to the text.
        
        obs = super()._set_text_vec(*args, **kwargs)
        if "text_vec" in obs:
            obs["text_vec"] = self._add_start_end_tokens(obs["text_vec"], True, True)

    def vectorize_fixed_candidates(self, *args, **kwargs):
         Add the start and end token when computing the candidate encodings
            in interactive mode.

After Change

    def _set_text_vec(self, *args, **kwargs):
         Add the start and end token to the text.
        
        obs = super()._set_text_vec(*args, **kwargs)
        if "text_vec" in obs and "added_start_end_tokens" not in obs:
            obs.force_set(
                "text_vec", self._add_start_end_tokens(obs["text_vec"], True, True)
            )
            obs["added_start_end_tokens"] = True
        return obs

    def vectorize_fixed_candidates(self, *args, **kwargs):
         Add the start and end token when computing the candidate encodings

Instances