for i, id in enumerate(s["id"].data):
src = input["src_tokens"].data[i, :]
// remove padding from ref
ref = utils.strip_pad(s["target"].data[i, :], self.pad)
yield id, src, ref, hypos[i]
def generate(self, src_tokens, src_lengths, beam_size=None, maxlen=None, prefix_tokens=None):
Generate a batch of translations.
After Change
for i, id in enumerate(s["id"].data):
src = input["src_tokens"].data[i, :]
// remove padding from ref
ref = utils.strip_pad(s["target"].data[i, :], self.pad) if s["target"] is not None else None
yield id, src, ref, hypos[i]
def generate(self, src_tokens, src_lengths, beam_size=None, maxlen=None, prefix_tokens=None):
Generate a batch of translations.
