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) _act_from_actor() (omnisafe.algorithms.model_based.planner.ARCPlanner method) _act_from_last_gaus() (omnisafe.algorithms.model_based.planner.ARCPlanner method) (omnisafe.algorithms.model_based.planner.CEMPlanner method) _algo_reset() (omnisafe.algorithms.model_based.base.PETS method) _alpha (omnisafe.algorithms.off_policy.SAC property) _alpha_discount() (omnisafe.algorithms.model_based.base.LOOP method) _augment_obs() (omnisafe.adapter.SauteAdapter method) _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) _compute_cost() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) _compute_reward() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) _compute_terminal() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel 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) _dist_xy() (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) _distribution() (omnisafe.models.actor.GaussianLearningActor method) (omnisafe.models.actor.GaussianSACActor method) (omnisafe.models.actor.PerturbationActor method) (omnisafe.models.actor.VAE method) (omnisafe.models.base.Actor method) _ego_xy() (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) _evaluation_single_step() (omnisafe.algorithms.model_based.base.PETS method) _fvp() (omnisafe.algorithms.on_policy.NaturalPG method) _get_coordinate_sensor() (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) _get_decay_loss() (omnisafe.algorithms.model_based.base.ensemble.EnsembleModel method) _get_flat_coordinate() (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) _init() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.DDPGLag method) (omnisafe.algorithms.off_policy.SAC method) (omnisafe.algorithms.off_policy.SACLag method) (omnisafe.algorithms.off_policy.TD3Lag method) (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.model_based.base.PETS method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.on_policy.PolicyGradient method) _init_log() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) (omnisafe.algorithms.model_based.CAPPETS method) (omnisafe.algorithms.model_based.CCEPETS method) (omnisafe.algorithms.model_based.RCEPETS method) (omnisafe.algorithms.model_based.SafeLOOP method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.DDPGLag method) (omnisafe.algorithms.off_policy.SAC method) (omnisafe.algorithms.off_policy.SACLag method) (omnisafe.algorithms.off_policy.TD3Lag method) (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.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) (omnisafe.algorithms.model_based.CAPPETS method) (omnisafe.algorithms.model_based.CCEPETS method) (omnisafe.algorithms.model_based.RCEPETS method) (omnisafe.algorithms.model_based.SafeLOOP method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.SAC method) (omnisafe.algorithms.off_policy.TD3 method) (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.ModelBasedAdapter method) (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) (omnisafe.adapter.SauteAdapter method) _log_value() (omnisafe.adapter.ModelBasedAdapter method) (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) (omnisafe.adapter.SauteAdapter method) _log_when_not_update() (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.DDPGLag method) (omnisafe.algorithms.off_policy.SAC method) (omnisafe.algorithms.off_policy.SACLag method) (omnisafe.algorithms.off_policy.TD3Lag method) _loss_pi() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.DDPGLag method) (omnisafe.algorithms.off_policy.SAC method) (omnisafe.algorithms.off_policy.SACLag method) (omnisafe.algorithms.off_policy.TD3Lag method) (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) _obs_lidar_pseudo() (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) _predict() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) _push() (omnisafe.common.normalizer.Normalizer method) _register() (omnisafe.envs.core.EnvRegister method) _reset_log() (omnisafe.adapter.ModelBasedAdapter method) (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) (omnisafe.adapter.SauteAdapter method) _safety_reward() (omnisafe.adapter.SauteAdapter method) _safety_step() (omnisafe.adapter.SauteAdapter method) _save_best() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) _save_model() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) (omnisafe.algorithms.model_based.CAPPETS method) _search_step_size() (omnisafe.algorithms.on_policy.TRPO method) _select_action() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) _select_elites() (omnisafe.algorithms.model_based.planner.ARCPlanner method) (omnisafe.algorithms.model_based.planner.CAPPlanner method) (omnisafe.algorithms.model_based.planner.CCEPlanner method) (omnisafe.algorithms.model_based.planner.CEMPlanner method) (omnisafe.algorithms.model_based.planner.RCEPlanner method) (omnisafe.algorithms.model_based.planner.SafeARCPlanner method) _state_action_repeat() (omnisafe.algorithms.model_based.planner.ARCPlanner method) (omnisafe.algorithms.model_based.planner.CEMPlanner method) _store_real_data() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) _update() (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.DDPGLag method) (omnisafe.algorithms.off_policy.SACLag method) (omnisafe.algorithms.off_policy.TD3Lag method) (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.model_based.base.LOOP method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.SAC method) (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.model_based.base.LOOP method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.on_policy.PolicyGradient method) _update_current_row() (omnisafe.common.logger.Logger method) _update_dynamics_model() (omnisafe.algorithms.model_based.base.PETS method) _update_epoch() (omnisafe.algorithms.model_based.base.PETS method) (omnisafe.algorithms.model_based.CAPPETS method) _update_mean_var() (omnisafe.algorithms.model_based.planner.ARCPlanner method) (omnisafe.algorithms.model_based.planner.CEMPlanner method) (omnisafe.algorithms.model_based.planner.SafeARCPlanner method) _update_policy() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.model_based.base.PETS method) _update_reward_critic() (omnisafe.algorithms.model_based.base.LOOP method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.off_policy.SAC method) (omnisafe.algorithms.off_policy.TD3 method) (omnisafe.algorithms.on_policy.PolicyGradient method) _variants() (omnisafe.common.experiment_grid.ExperimentGrid method) (omnisafe.common.statistics_tools.StatisticsTools method) _wrapper() (omnisafe.adapter.ModelBasedAdapter method) (omnisafe.adapter.OnlineAdapter method) (omnisafe.adapter.SauteAdapter method) A act() (omnisafe.common.simmer_agent.BaseSimmerAgent method) (omnisafe.common.simmer_agent.SimmerPIDAgent method) action_space (omnisafe.adapter.OfflineAdapter property) (omnisafe.adapter.OnlineAdapter property) (omnisafe.envs.core.CMDP property) ActionRepeat (class in omnisafe.envs.wrapper) 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) ARCPlanner (class in omnisafe.algorithms.model_based.planner) 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) BaseSimmerAgent (class in omnisafe.common.simmer_agent) 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 CAPPETS (class in omnisafe.algorithms.model_based) CAPPlanner (class in omnisafe.algorithms.model_based.planner) CCEPETS (class in omnisafe.algorithms.model_based) CCEPlanner (class in omnisafe.algorithms.model_based.planner) CEMPlanner (class in omnisafe.algorithms.model_based.planner) check_all_configs() (in module omnisafe.utils.config) check_variant_vaild() (omnisafe.common.experiment_grid.ExperimentGrid method) close() (omnisafe.adapter.OnlineAdapter method) (omnisafe.common.logger.Logger method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.custom_env.CustomEnv method) (omnisafe.envs.mujoco_env.MujocoEnv method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) CMDP (class in omnisafe.envs.core) collect() (omnisafe.common.offline.OfflineDataCollector method) combine() (omnisafe.common.statistics_tools.StatisticsTools method) 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) control_budget() (omnisafe.adapter.SimmerAdapter method) 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) CustomEnv (class in omnisafe.envs.custom_env) D DDPG (class in omnisafe.algorithms.off_policy) DDPGLag (class in omnisafe.algorithms.off_policy) decode() (omnisafe.models.actor.VAE method) decompress_key() (omnisafe.common.statistics_tools.StatisticsTools method) device (omnisafe.common.buffer.BaseBuffer property) dict2config() (omnisafe.utils.config.Config static method) dict_permutations() (omnisafe.common.statistics_tools.StatisticsTools 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) draw_graph() (omnisafe.common.statistics_tools.StatisticsTools method) draw_picture() (omnisafe.algorithms.model_based.base.PETS method) dump_tabular() (omnisafe.common.logger.Logger method) E encode() (omnisafe.models.actor.VAE method) ensemble_model (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel property) EnsembleDynamicsModel (class in omnisafe.algorithms.model_based.base.ensemble) EnsembleFC (class in omnisafe.algorithms.model_based.base.ensemble) EnsembleModel (class in omnisafe.algorithms.model_based.base.ensemble) entropy() (omnisafe.utils.math.TanhNormal method) env_spec_keys (omnisafe.adapter.OnlineAdapter property) EnvRegister (class in omnisafe.envs.core) eval_policy() (omnisafe.adapter.OffPolicyAdapter method) evaluate() (omnisafe.adapter.OfflineAdapter method) (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) fit() (omnisafe.algorithms.model_based.base.ensemble.StandardScaler method) FOCOPS (class in omnisafe.algorithms.on_policy) fork() (in module omnisafe.utils.distributed) forward() (omnisafe.algorithms.model_based.base.ensemble.EnsembleFC method) (omnisafe.algorithms.model_based.base.ensemble.EnsembleModel method) (omnisafe.common.normalizer.Normalizer method) (omnisafe.models.actor.GaussianLearningActor method) (omnisafe.models.actor.GaussianSACActor method) (omnisafe.models.actor.PerturbationActor method) (omnisafe.models.actor.VAE 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) (omnisafe.models.offline.dice.ObsEncoder method) forward_idx() (omnisafe.algorithms.model_based.base.ensemble.EnsembleModel 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_all_datasets() (omnisafe.utils.plotter.Plotter method) get_class() (omnisafe.envs.core.EnvRegister method) get_compressed_key() (omnisafe.common.statistics_tools.StatisticsTools method) get_cost_from_obs_tensor() (omnisafe.adapter.ModelBasedAdapter method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) get_datasets() (omnisafe.utils.plotter.Plotter 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_greedy_action() (omnisafe.common.simmer_agent.BaseSimmerAgent method) (omnisafe.common.simmer_agent.SimmerPIDAgent method) get_lidar_from_coordinate() (omnisafe.adapter.ModelBasedAdapter method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) get_rank() (in module omnisafe.utils.distributed) get_stats() (omnisafe.common.logger.Logger method) get_transpose() (in module omnisafe.utils.math) I imagine() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) init_weights() (in module omnisafe.algorithms.model_based.base.ensemble) 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.model_based.base.PETS method) (omnisafe.algorithms.off_policy.DDPG method) (omnisafe.algorithms.on_policy.PolicyGradient method) load_source() (omnisafe.common.statistics_tools.StatisticsTools 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.actor.PerturbationActor method) (omnisafe.models.actor.VAE method) (omnisafe.models.base.Actor method) Logger (class in omnisafe.common.logger) LOOP (class in omnisafe.algorithms.model_based.base) loss() (omnisafe.algorithms.model_based.base.ensemble.EnsembleModel method) (omnisafe.models.actor.VAE method) M make() (in module omnisafe.envs.core) make_config_groups() (omnisafe.common.statistics_tools.StatisticsTools method) make_plots() (omnisafe.utils.plotter.Plotter method) max_episode_steps (omnisafe.envs.core.CMDP property) (omnisafe.envs.custom_env.CustomEnv property) (omnisafe.envs.mujoco_env.MujocoEnv property) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv property) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased property) max_size (omnisafe.common.buffer.OffPolicyBuffer property) mean (omnisafe.common.normalizer.Normalizer property) (omnisafe.utils.math.TanhNormal property) metadata (omnisafe.envs.core.CMDP property) ModelBasedAdapter (class in omnisafe.adapter) ModelConfig (class in omnisafe.utils.config) MujocoEnv (class in omnisafe.envs.mujoco_env) 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) num_models (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel property) O ObsEncoder (class in omnisafe.models.offline.dice) observation_space (omnisafe.adapter.OfflineAdapter property) (omnisafe.adapter.OnlineAdapter property) (omnisafe.adapter.SauteAdapter property) (omnisafe.envs.core.CMDP property) ObsNormalize (class in omnisafe.envs.wrapper) OfflineAdapter (class in omnisafe.adapter) OfflineDataCollector (class in omnisafe.common.offline) OfflineDataset (class in omnisafe.common.offline) OfflineDatasetWithInit (class in omnisafe.common.offline) 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) output_action() (omnisafe.algorithms.model_based.planner.ARCPlanner method) (omnisafe.algorithms.model_based.planner.CEMPlanner method) (omnisafe.algorithms.model_based.planner.SafeARCPlanner method) P P3O (class in omnisafe.algorithms.on_policy) PCPO (class in omnisafe.algorithms.on_policy) PerturbationActor (class in omnisafe.models.actor) PETS (class in omnisafe.algorithms.model_based.base) phi (omnisafe.models.actor.PerturbationActor property) pid_update() (omnisafe.common.pid_lagrange.PIDLagrangian method) PIDLagrangian (class in omnisafe.common.pid_lagrange) plot_data() (omnisafe.utils.plotter.Plotter method) Plotter (class in omnisafe.utils.plotter) 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.actor.PerturbationActor method) (omnisafe.models.actor.VAE method) (omnisafe.models.base.Actor method) print() (omnisafe.common.experiment_grid.ExperimentGrid method) Q QCritic (class in omnisafe.models.critic) R RCEPETS (class in omnisafe.algorithms.model_based) RCEPlanner (class in omnisafe.algorithms.model_based.planner) recurisve_update() (omnisafe.utils.config.Config method) recursive_check_config() (in module omnisafe.utils.tools) register() (omnisafe.envs.core.EnvRegister method) register_agent() (omnisafe.common.offline.OfflineDataCollector method) register_key() (omnisafe.common.logger.Logger method) render() (omnisafe.adapter.ModelBasedAdapter method) (omnisafe.common.experiment_grid.ExperimentGrid method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.custom_env.CustomEnv method) (omnisafe.envs.mujoco_env.MujocoEnv method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) reset() (omnisafe.adapter.OfflineAdapter method) (omnisafe.adapter.OnlineAdapter method) (omnisafe.adapter.SauteAdapter method) (omnisafe.adapter.SimmerAdapter method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.custom_env.CustomEnv method) (omnisafe.envs.mujoco_env.MujocoEnv method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased 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.ModelBasedAdapter method) (omnisafe.adapter.OffPolicyAdapter method) (omnisafe.adapter.OnPolicyAdapter method) run() (omnisafe.common.experiment_grid.ExperimentGrid method) S SAC (class in omnisafe.algorithms.off_policy) SACLag (class in omnisafe.algorithms.off_policy) SafeARCPlanner (class in omnisafe.algorithms.model_based.planner) SafeLOOP (class in omnisafe.algorithms.model_based) SafeTanhTransformer (class in omnisafe.utils.math) SafetyGymnasiumEnv (class in omnisafe.envs.safety_gymnasium_env) SafetyGymnasiumModelBased (class in omnisafe.envs.safety_gymnasium_modelbased) sample() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) (omnisafe.common.offline.OfflineDataset method) (omnisafe.common.offline.OfflineDatasetWithInit method) sample_batch() (omnisafe.common.buffer.OffPolicyBuffer method) (omnisafe.common.buffer.VectorOffPolicyBuffer method) SauteAdapter (class in omnisafe.adapter) 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.custom_env.CustomEnv method) (omnisafe.envs.mujoco_env.MujocoEnv method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) setup_distributed() (in module omnisafe.utils.distributed) setup_torch_saver() (omnisafe.common.logger.Logger method) shape (omnisafe.common.normalizer.Normalizer property) SimmerAdapter (class in omnisafe.adapter) SimmerPIDAgent (class in omnisafe.common.simmer_agent) size (omnisafe.common.buffer.BaseBuffer property) (omnisafe.common.buffer.OffPolicyBuffer property) spec_log() (omnisafe.envs.custom_env.CustomEnv method) standardized_adv_c (omnisafe.common.buffer.OnPolicyBuffer property) standardized_adv_r (omnisafe.common.buffer.OnPolicyBuffer property) StandardScaler (class in omnisafe.algorithms.model_based.base.ensemble) state_size (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel property) StatisticsTools (class in omnisafe.common.statistics_tools) 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.OfflineAdapter method) (omnisafe.adapter.OnlineAdapter method) (omnisafe.adapter.SauteAdapter method) (omnisafe.envs.core.CMDP method) (omnisafe.envs.core.Wrapper method) (omnisafe.envs.custom_env.CustomEnv method) (omnisafe.envs.mujoco_env.MujocoEnv method) (omnisafe.envs.safety_gymnasium_env.SafetyGymnasiumEnv method) (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased method) (omnisafe.envs.wrapper.ActionRepeat 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) task (omnisafe.envs.safety_gymnasium_modelbased.SafetyGymnasiumModelBased property) TD3 (class in omnisafe.algorithms.off_policy) TD3Lag (class in omnisafe.algorithms.off_policy) 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) train() (omnisafe.algorithms.model_based.base.ensemble.EnsembleDynamicsModel method) train_ensemble() (omnisafe.algorithms.model_based.base.ensemble.EnsembleModel method) transform() (omnisafe.algorithms.model_based.base.ensemble.StandardScaler method) TRPO (class in omnisafe.algorithms.on_policy) TRPOLag (class in omnisafe.algorithms.on_policy) U unbatched_forward() (in module omnisafe.algorithms.model_based.base.ensemble) unregister() (omnisafe.envs.core.EnvRegister method) Unsqueeze (class in omnisafe.envs.wrapper) update_dict() (in module omnisafe.utils.tools) (omnisafe.common.experiment_grid.ExperimentGrid method) (omnisafe.common.statistics_tools.StatisticsTools method) update_lagrange_multiplier() (omnisafe.common.lagrange.Lagrange method) V VAE (class in omnisafe.models.actor) 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) (omnisafe.common.statistics_tools.StatisticsTools 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)