in agents/ql_diffusion.py [0:0]
def step_ema(self): if self.step < self.step_start_ema: return self.ema.update_model_average(self.ema_model, self.actor)