def __call__(self, x: torch.Tensor):
    x = self.src_embed(x)
    res = self.transformer(x)
    return (self.generator(res), None)
def __call__(self, x: torch.Tensor):
    x = self.src_embed(x)
    res = self.transformer(x)
    return (self.generator(res), None)