              mask = partial(face.get_landmark_mask,
                             size,
                             area,
                             aligned=True,
                             centering=centering,
                             dilation=size // 32,
                             blur_kernel=size // 16,
                             as_zip=True)
              for filename in self._hash_to_filenames(side, fhash):
                  masks[filename] = mask
 
</s> remove         if self._alignments_version[side] == 1.0:
            centering = "legacy"
            size = self._training_sizes[side]
        else:
            centering = self._config["centering"]
            size = list(self._aligned_faces[side].values())[0].get_cropped_size(centering)
 </s> add         size = list(self._aligned_faces[side].values())[0].size </s> remove             size = self._training_sizes[side]
            centering = "legacy" if self._alignments_version[side] == 1.0 else "head"
            logger.debug("side: %s, centering: %s", side, centering)
 </s> add  </s> remove                 face.load_aligned(None, size=size, centering=centering)
 </s> add                 face.aligned = AlignedFace(face.landmarks_xy,
                                           centering=self._config["centering"],
                                           size=size,
                                           is_aligned=True) </s> remove         cropped = np.zeros((batch.shape[0], size, size, batch.shape[3]), dtype=batch.dtype)

        for out, align, img in zip(cropped, aligned, batch):
            slices = align.get_cropped_slices(self._config["centering"])
            out[slices["out"][0], slices["out"][1], :] = img[slices["in"][0], slices["in"][1], :]
 </s> add         cropped = np.array([align.extract_face(img) for align, img in zip(aligned, batch)]) </s> remove         crop_size = 2 * int(np.rint(src_size / (1 - _EXTRACT_RATIOS["legacy"]) / 2))
 </s> add         crop_size = get_centered_size("face", "legacy", self.stored_size)
                           centering=self._config["centering"],
              mask = partial(face.get_landmark_mask,
                             size,
                             area,
                             aligned=True,
                           centering=self._config["centering"],
                             dilation=size // 32,
                             blur_kernel=size // 16,
                             as_zip=True)
              for filename in self._hash_to_filenames(side, fhash):
                  masks[filename] = mask
keep keep keep keep replace keep keep keep keep keep
 Bugfix - Warp to landmarks 
plugins/train/trainer/_base.py