    def forward_target_critic(
        self,
        observations: Data,
        actions: jax.Array,
        rng: PRNGKey,
    ) -> jax.Array:
        """
        Forward pass for target critic network.
        Pass grad_params to use non-default parameters (e.g. for gradients).
        """
        return self.forward_critic(
            observations, actions, rng=rng, grad_params=self.state.target_params
        )
Forward pass for target critic network.
Pass grad_params to use non-default parameters (e.g. for gradients).
serl_launcher/serl_launcher/agents/continuous/sac.py