code: def post_optimization_step(self, batch_info, device, model, rollout):
          
          if batch_info.aggregate_batch_number % self.target_update_frequency == 0:
              self.target_model.load_state_dict(model.state_dict())
              self.target_model.eval()
docs: Steps to take after optimization has been done