Method generate_src_mask

mogen/models/transformers/intergen.py:158–164

(self, T, length)

        return losses["total"], losses

    def generate_src_mask(self, T, length):
        B = length.shape[0]
        src_mask = torch.ones(B, T)
        for i in range(B):
            for j in range(length[i], T):
                src_mask[i, j] = 0
        return src_mask

    def encode_motion(self,
                      motion,