in sparse_autoencoder/train.py [0:0]
def step(self): torch._foreach_lerp_( self.ema_weights, list(self.model.parameters()), 1 - self.ema_multiplier, ) self.ema_steps += 1