def step_ema()

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)