code: def post_optimization_step(self, batch_info, device, model, rollout):
          
          # Update target model
          for model_param, target_param in zip(model.parameters(), self.target_model.parameters()):
              # EWMA average model update
              target_param.data.mul_(1 - self.tau).add_(model_param.data * self.tau)
docs: Steps to take after optimization has been done