Index _ | A | B | C | D | E | F | G | I | L | M | N | O | P | Q | R | S | T | U | V | W _ __check_algo_configs() (in module omnisafe.utils.config) __check_logger_configs() (in module omnisafe.utils.config) _calculate_adv_and_value_targets() (omnisafe.common.buffer.OnPolicyBuffer method) _calculate_v_trace() (omnisafe.common.buffer.OnPolicyBuffer static method) _compute_adv_surrogate() (omnisafe.algorithms.on_policy.FOCOPS method) (omnisafe.algorithms.on_policy.IPO method) (omnisafe.algorithms.on_policy.OnCRPO method) (omnisafe.algorithms.on_policy.PolicyGradient method) (omnisafe.algorithms.on_policy.PPOLag method) (omnisafe.algorithms.on_policy.TRPOLag method) _cpo_search_step() (omnisafe.algorithms.on_policy.CPO method) _default_shorthand() (omnisafe.common.experiment_grid.ExperimentGrid method) _determine_case() (omnisafe.algorithms.on_policy.CPO method) _distribution() (omnisafe.models.actor.GaussianLearningActor method) (omnisafe.models.actor.GaussianSACActor method) (omnisafe.models.base.Actor method) _fvp() (omnisafe.algorithms.on_policy.NaturalPG method) _init() (omnisafe.algorithms.on_policy.CUP method) (omnisafe.algorithms.on_policy.FOCOPS method) (omnisafe.algorithms.on_policy.PolicyGradient method) (omnisafe.algorithms.on_policy.PPOLag method) (omnisafe.algorithms.on_policy.TRPOLag method) _init_env() (omnisafe.algorithms.on_policy.PolicyGradient method) _init_log() (omnisafe.algorithms.on_policy.CPO method) (omnisafe.algorithms.on_policy.CUP method) (omnisafe.algorithms.on_policy.FOCOPS method) (omnisafe.algorithms.on_policy.IPO method) (omnisafe.algorithms.on_policy.NaturalPG method) (omnisafe.algorithms.on_policy.OnCRPO method) (omnisafe.algorithms.on_policy.P3O method) (omnisafe.algorithms.on_policy.PolicyGradient method) (omnisafe.algorithms.on_policy.PPOLag method) (omnisafe.algorithms.on_policy.TRPO method) (omnisafe.algorithms.on_policy.TRPOLag method) _init_model() (omnisafe.algorithms.on_policy.PolicyGradient method) _init_statistical_tools() (omnisafe.common.experiment_grid.ExperimentGrid method) _inverse() (omnisafe.utils.math.SafeTanhTransformer method) _log_metrics() (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) _log_value() (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) _loss_pi() (omnisafe.algorithms.on_policy.FOCOPS method) (omnisafe.algorithms.on_policy.PolicyGradient method) (omnisafe.algorithms.on_policy.PPO method) _loss_pi_cost() (omnisafe.algorithms.on_policy.CPO method) (omnisafe.algorithms.on_policy.CUP method) (omnisafe.algorithms.on_policy.P3O method) _push() (omnisafe.common.normalizer.Normalizer method) _register() (omnisafe.envs.core.EnvRegister method) _reset_log() (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) _search_step_size() (omnisafe.algorithms.on_policy.TRPO method) _update() (omnisafe.algorithms.on_policy.CUP method) (omnisafe.algorithms.on_policy.FOCOPS method) (omnisafe.algorithms.on_policy.NaturalPG method) (omnisafe.algorithms.on_policy.PolicyGradient method) (omnisafe.algorithms.on_policy.PPOLag method) (omnisafe.algorithms.on_policy.TRPOLag method) _update_actor() (omnisafe.algorithms.on_policy.CPO method) (omnisafe.algorithms.on_policy.NaturalPG method) (omnisafe.algorithms.on_policy.P3O method) (omnisafe.algorithms.on_policy.PCPO method) (omnisafe.algorithms.on_policy.PolicyGradient method) (omnisafe.algorithms.on_policy.TRPO method) _update_cost_critic() (omnisafe.algorithms.on_policy.PolicyGradient method) _update_current_row() (omnisafe.common.logger.Logger method) _update_reward_critic() (omnisafe.algorithms.on_policy.PolicyGradient method) _variants() (omnisafe.common.experiment_grid.ExperimentGrid method) _wrapper() (omnisafe.adapter.OnlineAdapter method) A action_space (omnisafe.adapter.OnlineAdapter property) (omnisafe.envs.core.CMDP property) ActionScale (class in omnisafe.envs.wrapper) Actor (class in omnisafe.models.base) ActorBuilder (class in omnisafe.models.actor) ActorCritic (class in omnisafe.models.actor_critic) ActorQCritic (class in omnisafe.models.actor_critic) add() (omnisafe.common.experiment_grid.ExperimentGrid method) add_field() (omnisafe.common.buffer.BaseBuffer method) (omnisafe.common.buffer.VectorOffPolicyBuffer method) all_bools() (in module omnisafe.utils.exp_grid_tools) analyze() (omnisafe.common.experiment_grid.ExperimentGrid method) annealing() (omnisafe.models.actor_critic.ActorCritic method) AutoReset (class in omnisafe.envs.wrapper) avg_grads() (in module omnisafe.utils.distributed) avg_params() (in module omnisafe.utils.distributed) avg_tensor() (in module omnisafe.utils.distributed) B BaseBuffer (class in omnisafe.common.buffer) batch_size (omnisafe.common.buffer.OffPolicyBuffer property) build_actor() (omnisafe.models.actor.ActorBuilder method) build_critic() (omnisafe.models.critic.CriticBuilder method) build_mlp_network() (in module omnisafe.utils.model) C check_all_configs() (in module omnisafe.utils.config) check_variant_vaild() (omnisafe.common.experiment_grid.ExperimentGrid method) close() (omnisafe.common.logger.Logger method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) CMDP (class in omnisafe.envs.core) compute_lambda_loss() (omnisafe.common.lagrange.Lagrange method) Config (class in omnisafe.utils.config) conjugate_gradients() (in module omnisafe.utils.math) ConstraintActorCritic (class in omnisafe.models.actor_critic) ConstraintActorQCritic (class in omnisafe.models.actor_critic) CostNormalize (class in omnisafe.envs.wrapper) CPO (class in omnisafe.algorithms.on_policy) Critic (class in omnisafe.models.base) CriticBuilder (class in omnisafe.models.critic) CUP (class in omnisafe.algorithms.on_policy) current_epoch (omnisafe.common.logger.Logger property) custom_cfgs_to_dict() (in module omnisafe.utils.tools) D device (omnisafe.common.buffer.BaseBuffer property) dict2config() (omnisafe.utils.config.Config static method) discount_cumsum() (in module omnisafe.utils.math) dist_avg() (in module omnisafe.utils.distributed) dist_max() (in module omnisafe.utils.distributed) dist_min() (in module omnisafe.utils.distributed) dist_op() (in module omnisafe.utils.distributed) dist_statistics_scalar() (in module omnisafe.utils.distributed) dist_sum() (in module omnisafe.utils.distributed) dump_tabular() (omnisafe.common.logger.Logger method) E entropy() (omnisafe.utils.math.TanhNormal method) EnvRegister (class in omnisafe.envs.core) eval_policy() (omnisafe.adapter.OffPolicyAdapter method) evaluate() (omnisafe.common.experiment_grid.ExperimentGrid method) expand() (omnisafe.utils.math.TanhNormal method) ExperimentGrid (class in omnisafe.common.experiment_grid) F finish_path() (omnisafe.common.buffer.OnPolicyBuffer method) (omnisafe.common.buffer.VectorOnPolicyBuffer method) FOCOPS (class in omnisafe.algorithms.on_policy) fork() (in module omnisafe.utils.distributed) forward() (omnisafe.common.normalizer.Normalizer method) (omnisafe.models.actor.GaussianLearningActor method) (omnisafe.models.actor.GaussianSACActor method) (omnisafe.models.actor_critic.ActorCritic method) (omnisafe.models.actor_critic.ActorQCritic method) (omnisafe.models.actor_critic.ConstraintActorCritic method) (omnisafe.models.base.Actor method) (omnisafe.models.critic.QCritic method) (omnisafe.models.critic.VCritic method) G GaussianActor (class in omnisafe.models.actor) GaussianLearningActor (class in omnisafe.models.actor) GaussianSACActor (class in omnisafe.models.actor) get() (omnisafe.common.buffer.OnPolicyBuffer method) (omnisafe.common.buffer.VectorOnPolicyBuffer method) get_activation() (in module omnisafe.utils.model) get_class() (omnisafe.envs.core.EnvRegister method) get_default_kwargs_yaml() (in module omnisafe.utils.config) get_diagonal() (in module omnisafe.utils.math) get_flat_gradients_from() (in module omnisafe.utils.tools) get_flat_params_from() (in module omnisafe.utils.tools) get_rank() (in module omnisafe.utils.distributed) get_stats() (omnisafe.common.logger.Logger method) get_transpose() (in module omnisafe.utils.math) I initialize_layer() (in module omnisafe.utils.model) IPO (class in omnisafe.algorithms.on_policy) L Lagrange (class in omnisafe.common.lagrange) lagrangian_multiplier (omnisafe.common.pid_lagrange.PIDLagrangian property) learn() (omnisafe.algorithms.on_policy.PolicyGradient method) load_state_dict() (omnisafe.common.normalizer.Normalizer method) load_yaml() (in module omnisafe.utils.tools) loc (omnisafe.utils.math.TanhNormal property) log() (omnisafe.common.logger.Logger method) log_dir (omnisafe.common.logger.Logger property) log_prob() (omnisafe.models.actor.GaussianLearningActor method) (omnisafe.models.actor.GaussianSACActor method) (omnisafe.models.base.Actor method) Logger (class in omnisafe.common.logger) M make() (in module omnisafe.envs.core) max_size (omnisafe.common.buffer.OffPolicyBuffer property) mean (omnisafe.common.normalizer.Normalizer property) (omnisafe.utils.math.TanhNormal property) metadata (omnisafe.envs.core.CMDP property) ModelConfig (class in omnisafe.utils.config) N NaturalPG (class in omnisafe.algorithms.on_policy) normalize() (omnisafe.common.normalizer.Normalizer method) Normalizer (class in omnisafe.common.normalizer) num_buffers (omnisafe.common.buffer.VectorOnPolicyBuffer property) num_envs (omnisafe.common.buffer.VectorOffPolicyBuffer property) (omnisafe.envs.core.CMDP property) O observation_space (omnisafe.adapter.OnlineAdapter property) (omnisafe.envs.core.CMDP property) ObsNormalize (class in omnisafe.envs.wrapper) OffPolicyAdapter (class in omnisafe.adapter) OffPolicyBuffer (class in omnisafe.common.buffer) OnCRPO (class in omnisafe.algorithms.on_policy) OnlineAdapter (class in omnisafe.adapter) OnPolicyAdapter (class in omnisafe.adapter) OnPolicyBuffer (class in omnisafe.common.buffer) P P3O (class in omnisafe.algorithms.on_policy) PCPO (class in omnisafe.algorithms.on_policy) pid_update() (omnisafe.common.pid_lagrange.PIDLagrangian method) PIDLagrangian (class in omnisafe.common.pid_lagrange) PolicyGradient (class in omnisafe.algorithms.on_policy) polyak_update() (omnisafe.models.actor_critic.ActorQCritic method) (omnisafe.models.actor_critic.ConstraintActorQCritic method) PPO (class in omnisafe.algorithms.on_policy) PPOLag (class in omnisafe.algorithms.on_policy) predict() (omnisafe.models.actor.GaussianLearningActor method) (omnisafe.models.actor.GaussianSACActor method) (omnisafe.models.base.Actor method) print() (omnisafe.common.experiment_grid.ExperimentGrid method) Q QCritic (class in omnisafe.models.critic) R recurisve_update() (omnisafe.utils.config.Config method) recursive_check_config() (in module omnisafe.utils.tools) register() (omnisafe.envs.core.EnvRegister method) register_key() (omnisafe.common.logger.Logger method) render() (omnisafe.common.experiment_grid.ExperimentGrid method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) reset() (omnisafe.adapter.OnlineAdapter method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.wrapper.ObsNormalize method) (omnisafe.envs.wrapper.TimeLimit method) (omnisafe.envs.wrapper.Unsqueeze method) RewardNormalize (class in omnisafe.envs.wrapper) rollout() (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) run() (omnisafe.common.experiment_grid.ExperimentGrid method) S SafeTanhTransformer (class in omnisafe.utils.math) SafetyGymnasiumEnv (class in omnisafe.envs.safety_gymnasium_env) sample_action() (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) sample_batch() (omnisafe.common.buffer.OffPolicyBuffer method) (omnisafe.common.buffer.VectorOffPolicyBuffer method) save() (omnisafe.adapter.OnlineAdapter method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.wrapper.CostNormalize method) (omnisafe.envs.wrapper.ObsNormalize method) (omnisafe.envs.wrapper.RewardNormalize method) save_config() (omnisafe.common.logger.Logger method) save_grid_config() (omnisafe.common.experiment_grid.ExperimentGrid method) save_results() (omnisafe.common.experiment_grid.ExperimentGrid method) save_same_exps_config() (omnisafe.common.experiment_grid.ExperimentGrid method) scale (omnisafe.utils.math.TanhNormal property) seed_all() (in module omnisafe.utils.tools) set_annealing() (omnisafe.models.actor_critic.ActorCritic method) set_param_values_to_model() (in module omnisafe.utils.tools) set_seed() (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) setup_distributed() (in module omnisafe.utils.distributed) setup_torch_saver() (omnisafe.common.logger.Logger method) shape (omnisafe.common.normalizer.Normalizer property) size (omnisafe.common.buffer.BaseBuffer property) (omnisafe.common.buffer.OffPolicyBuffer property) standardized_adv_c (omnisafe.common.buffer.OnPolicyBuffer property) standardized_adv_r (omnisafe.common.buffer.OnPolicyBuffer property) std (omnisafe.common.normalizer.Normalizer property) (omnisafe.models.actor.GaussianActor property) (omnisafe.models.actor.GaussianLearningActor property) (omnisafe.models.actor.GaussianSACActor property) stddev (omnisafe.utils.math.TanhNormal property) step() (omnisafe.adapter.OnlineAdapter method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.wrapper.ActionScale method) (omnisafe.envs.wrapper.AutoReset method) (omnisafe.envs.wrapper.CostNormalize method) (omnisafe.envs.wrapper.ObsNormalize method) (omnisafe.envs.wrapper.RewardNormalize method) (omnisafe.envs.wrapper.TimeLimit method) (omnisafe.envs.wrapper.Unsqueeze method) (omnisafe.models.actor_critic.ActorCritic method) (omnisafe.models.actor_critic.ActorQCritic method) (omnisafe.models.actor_critic.ConstraintActorCritic method) store() (omnisafe.common.buffer.BaseBuffer method) (omnisafe.common.buffer.OffPolicyBuffer method) (omnisafe.common.buffer.OnPolicyBuffer method) (omnisafe.common.buffer.VectorOnPolicyBuffer method) (omnisafe.common.logger.Logger method) support_envs() (omnisafe.envs.core.CMDP class method) (omnisafe.envs.core.EnvRegister method) sync_params() (in module omnisafe.utils.distributed) T TanhNormal (class in omnisafe.utils.math) time_limit (omnisafe.envs.core.CMDP property) TimeLimit (class in omnisafe.envs.wrapper) todict() (omnisafe.utils.config.Config method) tojson() (omnisafe.utils.config.Config method) torch_save() (omnisafe.common.logger.Logger method) TRPO (class in omnisafe.algorithms.on_policy) TRPOLag (class in omnisafe.algorithms.on_policy) U Unsqueeze (class in omnisafe.envs.wrapper) update_dict() (in module omnisafe.utils.tools) (omnisafe.common.experiment_grid.ExperimentGrid method) update_lagrange_multiplier() (omnisafe.common.lagrange.Lagrange method) V valid_str() (in module omnisafe.utils.exp_grid_tools) variance (omnisafe.utils.math.TanhNormal property) variant_name() (omnisafe.common.experiment_grid.ExperimentGrid method) variants() (omnisafe.common.experiment_grid.ExperimentGrid method) VCritic (class in omnisafe.models.critic) VectorOffPolicyBuffer (class in omnisafe.common.buffer) VectorOnPolicyBuffer (class in omnisafe.common.buffer) W world_size() (in module omnisafe.utils.distributed) Wrapper (class in omnisafe.envs.core)