def METHOD_NAME(self):
    config = super().METHOD_NAME()
    config.update(
        {
            "augmentations_per_image": self.augmentations_per_image,
            "auto_vectorize": self.auto_vectorize,
            "rate": self.rate,
            "layers": self.layers,
            "seed": self.seed,
        }
    )
    return config