keras/src/callbacks/swap_ema_weights.py/SwapEMAWeights/__init__
class SwapEMAWeights:    def __init__(self, swap_on_epoch=False):
        super().__init__()
        self.swap_on_epoch = swap_on_epoch

        self._ema_weights_in_model = False