if self.predictor_tgt:
                model_out = self.predictor_tgt(batch, target_side=const.PE)
                logits = model_out[const.PE]
                outputs[const.PE] = logits