Index

_

__contains__() (protomotions.utils.config_builder.ConfigBuilder method) __getitem__() (protomotions.utils.config_builder.ConfigBuilder method) __init__() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.amp.config.AMPAgentConfig method) (protomotions.agents.amp.config.AMPModelConfig method) (protomotions.agents.amp.config.AMPParametersConfig method) (protomotions.agents.amp.config.DiscriminatorConfig method) (protomotions.agents.amp.model.AMPModel method) (protomotions.agents.amp.model.Discriminator method) (protomotions.agents.ase.agent.ASE method) (protomotions.agents.ase.config.ASEAgentConfig method) (protomotions.agents.ase.config.ASEDiscriminatorEncoderConfig method) (protomotions.agents.ase.config.ASEParametersConfig method) (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.base_agent.config.BaseAgentConfig method) (protomotions.agents.base_agent.config.BaseModelConfig method) (protomotions.agents.base_agent.config.MaxEpisodeLengthManagerConfig method) (protomotions.agents.base_agent.config.OptimizerConfig method) (protomotions.agents.base_agent.model.BaseModel method) (protomotions.agents.callbacks.slurm_autoresume_srun.AutoResumeCallbackSrun method) (protomotions.agents.common.common.Flatten method) (protomotions.agents.common.common.MultiInputModule method) (protomotions.agents.common.common.MultiOutputModule method) (protomotions.agents.common.common.NormObsBase method) (protomotions.agents.common.common.SequentialModule method) (protomotions.agents.common.config.FlattenConfig method) (protomotions.agents.common.config.MLPLayerConfig method) (protomotions.agents.common.config.MLPWithConcatConfig method) (protomotions.agents.common.config.ModuleOperationConfig method) (protomotions.agents.common.config.ModuleOperationExpandConfig method) (protomotions.agents.common.config.ModuleOperationForwardConfig method) (protomotions.agents.common.config.ModuleOperationPermuteConfig method) (protomotions.agents.common.config.ModuleOperationReshapeConfig method) (protomotions.agents.common.config.ModuleOperationSphereProjectionConfig method) (protomotions.agents.common.config.ModuleOperationSqueezeConfig method) (protomotions.agents.common.config.ModuleOperationUnsqueezeConfig method) (protomotions.agents.common.config.MultiInputModuleConfig method) (protomotions.agents.common.config.MultiOutputModuleConfig method) (protomotions.agents.common.config.NormObsBaseConfig method) (protomotions.agents.common.config.SequentialModuleConfig method) (protomotions.agents.common.config.TransformerConfig method) (protomotions.agents.common.mlp.MLPWithConcat method) (protomotions.agents.common.transformer.Transformer method) (protomotions.agents.evaluators.base_evaluator.BaseEvaluator method) (protomotions.agents.evaluators.base_evaluator.SmoothnessMetricPlugin method) (protomotions.agents.evaluators.config.EvaluatorConfig method) (protomotions.agents.evaluators.config.MimicEvaluatorConfig method) (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) (protomotions.agents.evaluators.MotionMetrics method) (protomotions.agents.evaluators.smoothness_evaluator.SmoothnessEvaluator method) (protomotions.agents.evaluators.SmoothnessEvaluator method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.masked_mimic.config.FeedForwardModelConfig method) (protomotions.agents.masked_mimic.config.KLDScheduleConfig method) (protomotions.agents.masked_mimic.config.MaskedMimicAgentConfig method) (protomotions.agents.masked_mimic.config.MaskedMimicModelConfig method) (protomotions.agents.masked_mimic.config.VaeConfig method) (protomotions.agents.masked_mimic.model.FeedForwardModel method) (protomotions.agents.masked_mimic.model.MaskedMimicModel method) (protomotions.agents.ppo.agent.PPO method) (protomotions.agents.ppo.config.AdvantageNormalizationConfig method) (protomotions.agents.ppo.config.PPOActorConfig method) (protomotions.agents.ppo.config.PPOAgentConfig method) (protomotions.agents.ppo.config.PPOModelConfig method) (protomotions.agents.ppo.model.PPOActor method) (protomotions.agents.ppo.model.PPOModel method) (protomotions.agents.utils.data.DictDataset method) (protomotions.agents.utils.data.ExperienceBuffer method) (protomotions.agents.utils.metering.TensorAverageMeter method) (protomotions.agents.utils.metering.TensorAverageMeterDict method) (protomotions.agents.utils.metering.Timer method) (protomotions.agents.utils.metering.TimeReport method) (protomotions.agents.utils.normalization.RewardRunningMeanStd method) (protomotions.agents.utils.normalization.RunningMeanStd method) (protomotions.agents.utils.replay_buffer.ReplayBuffer method) (protomotions.agents.utils.step_tracker.StepTracker method) (protomotions.robot_configs.amp.AMPRobotConfig method) (protomotions.robot_configs.base.RobotConfig method) (protomotions.robot_configs.g1.G1RobotConfig method) (protomotions.robot_configs.h1_2.H1_2RobotConfig method) (protomotions.robot_configs.smpl.SmplRobotConfig method) (protomotions.robot_configs.smplx.SMPLXRobotConfig method) (protomotions.utils.config_builder.ConfigBuilder method) (protomotions.utils.export_utils.ONNXExportWrapper method) (protomotions.utils.fabric_config.FabricConfig method)

__post_init__() (protomotions.robot_configs.base.RobotConfig method) _actor (protomotions.agents.amp.model.AMPModel attribute) (protomotions.agents.ppo.model.PPOModel attribute) _critic (protomotions.agents.amp.model.AMPModel attribute) (protomotions.agents.ppo.model.PPOModel attribute) _discriminator (protomotions.agents.amp.model.AMPModel attribute) _head (protomotions.agents.utils.replay_buffer.ReplayBuffer attribute) _is_full (protomotions.agents.utils.replay_buffer.ReplayBuffer attribute)

A

accelerator (protomotions.utils.fabric_config.FabricConfig attribute) activation (protomotions.agents.common.config.MLPLayerConfig attribute) (protomotions.agents.common.config.TransformerConfig attribute) actor (protomotions.agents.ppo.agent.PPO attribute) (protomotions.agents.ppo.config.PPOModelConfig attribute) actor_clip_frac_threshold (protomotions.agents.ppo.config.PPOAgentConfig attribute) actor_logstd (protomotions.agents.ppo.config.PPOActorConfig attribute) actor_optimizer (protomotions.agents.ppo.config.PPOModelConfig attribute) actor_step() (protomotions.agents.ppo.agent.PPO method) add() (protomotions.agents.utils.metering.TensorAverageMeter method) (protomotions.agents.utils.metering.TensorAverageMeterDict method) add_agent_info_to_obs() (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) add_extra_obs_to_agent() (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) add_timer() (protomotions.agents.utils.metering.TimeReport method) advance() (protomotions.agents.utils.step_tracker.StepTracker method) advantage_normalization (protomotions.agents.ppo.config.PPOAgentConfig attribute) AdvantageNormalizationConfig (class in protomotions.agents.ppo.config) aggregate_scalar_metrics() (in module protomotions.agents.utils.training) all_discriminator_weights() (protomotions.agents.amp.model.Discriminator method) (protomotions.agents.ase.model.ASEDiscriminatorEncoder method)

all_enc_weights() (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) all_weights() (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) alternative_quat_angle_axis() (protomotions.utils.rotations.TestQuatAngleAxis method) AMP (class in protomotions.agents.amp.agent) amp_parameters (protomotions.agents.amp.config.AMPAgentConfig attribute) amp_replay_buffer (protomotions.agents.amp.agent.AMP attribute) AMPAgentConfig (class in protomotions.agents.amp.config) AMPModel (class in protomotions.agents.amp.model) AMPModelConfig (class in protomotions.agents.amp.config) AMPParametersConfig (class in protomotions.agents.amp.config) AMPRobotConfig (class in protomotions.robot_configs.amp) apply_all_inference_overrides() (in module protomotions.utils.inference_utils) apply_backward_compatibility_fixes() (in module protomotions.utils.inference_utils) apply_config_overrides() (in module protomotions.utils.config_utils) apply_module_operations() (in module protomotions.agents.common.common) as_mesh() (in module protomotions.utils.mesh_utils) ASE (class in protomotions.agents.ase.agent) ase_parameters (protomotions.agents.ase.config.ASEAgentConfig attribute) ASEAgentConfig (class in protomotions.agents.ase.config) ASEDiscriminatorEncoder (class in protomotions.agents.ase.model) ASEDiscriminatorEncoderConfig (class in protomotions.agents.ase.config) ASEParametersConfig (class in protomotions.agents.ase.config) asset (protomotions.robot_configs.base.RobotConfig attribute) AutoResumeCallbackSrun (class in protomotions.agents.callbacks.slurm_autoresume_srun)

B

BaseAgent (class in protomotions.agents.base_agent.agent) BaseAgentConfig (class in protomotions.agents.base_agent.config) BaseEnv (built-in class) BaseEvaluator (class in protomotions.agents.evaluators.base_evaluator) BaseModel (class in protomotions.agents.base_agent.model) BaseModelConfig (class in protomotions.agents.base_agent.config) batch_size (protomotions.agents.base_agent.config.BaseAgentConfig attribute) batch_update_data() (protomotions.agents.utils.data.ExperienceBuffer method) before_play_steps() (protomotions.agents.callbacks.slurm_autoresume_srun.AutoResumeCallbackSrun method) betas (protomotions.agents.base_agent.config.OptimizerConfig attribute)

bounds_loss() (in module protomotions.agents.utils.training) bounds_loss_coef (protomotions.agents.base_agent.config.BaseAgentConfig attribute) build_all_components() (in module protomotions.utils.component_builder) build_mlp() (in module protomotions.agents.common.mlp) build_motion_lib_from_config() (in module protomotions.utils.component_builder) build_norm() (protomotions.agents.common.common.NormObsBase method) build_scene_lib_from_config() (in module protomotions.utils.component_builder) build_simulator_from_config() (in module protomotions.utils.component_builder) build_standard_configs() (in module protomotions.utils.config_builder) build_terrain_from_config() (in module protomotions.utils.component_builder)

C

calc_frame_blend() (in module protomotions.utils.motion_interpolation_utils) calc_von_mises_fisher_enc_error() (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) calculate_extra_actor_loss() (protomotions.agents.ase.agent.ASE method) (protomotions.agents.ppo.agent.PPO method) calculate_extra_loss() (protomotions.agents.masked_mimic.agent.MaskedMimic method) callbacks (protomotions.utils.fabric_config.FabricConfig attribute) check_grad_mag (protomotions.agents.base_agent.config.BaseAgentConfig attribute) check_obs_for_nans() (protomotions.agents.base_agent.agent.BaseAgent method) clamp_range (protomotions.agents.ppo.config.AdvantageNormalizationConfig attribute) cleanup_after_evaluation() (protomotions.agents.evaluators.base_evaluator.BaseEvaluator method) (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) clear() (protomotions.agents.utils.metering.TensorAverageMeter method) (protomotions.agents.utils.metering.TensorAverageMeterDict method) (protomotions.agents.utils.metering.Timer method) clear_timer() (protomotions.agents.utils.metering.TimeReport method) clip_critic_loss (protomotions.agents.ppo.config.PPOAgentConfig attribute) collect_rollout_step() (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) combine_moments() (in module protomotions.agents.utils.normalization) common_naming_to_robot_body_names (protomotions.robot_configs.base.RobotConfig attribute) compute() (protomotions.agents.evaluators.base_evaluator.SmoothnessMetricPlugin method) compute_bounding_box() (in module protomotions.utils.mesh_utils) compute_disc_reward() (protomotions.agents.amp.model.Discriminator method) compute_finite_difference_jitter_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) compute_jitter_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) compute_mi_reward() (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) compute_neg_acc() (protomotions.agents.amp.agent.AMP static method) compute_normalized_jerk_from_pos() (protomotions.agents.evaluators.smoothness_evaluator.SmoothnessEvaluator method) (protomotions.agents.evaluators.SmoothnessEvaluator method) compute_pos_acc() (protomotions.agents.amp.agent.AMP static method) compute_rotation_jitter_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) compute_smoothness_metrics() (protomotions.agents.evaluators.smoothness_evaluator.SmoothnessEvaluator method) (protomotions.agents.evaluators.SmoothnessEvaluator method) compute_uniformity_loss() (protomotions.agents.ase.agent.ASE method)

conditional_discriminator (protomotions.agents.amp.config.AMPParametersConfig attribute) config (protomotions.agents.amp.agent.AMP attribute) (protomotions.agents.amp.model.AMPModel attribute) (protomotions.agents.amp.model.Discriminator attribute) (protomotions.agents.ase.agent.ASE attribute) (protomotions.agents.ase.model.ASEDiscriminatorEncoder attribute) (protomotions.agents.base_agent.model.BaseModel attribute) (protomotions.agents.common.common.Flatten attribute) (protomotions.agents.common.common.MultiInputModule attribute) (protomotions.agents.common.common.MultiOutputModule attribute) (protomotions.agents.common.common.SequentialModule attribute) (protomotions.agents.common.mlp.MLPWithConcat attribute) (protomotions.agents.masked_mimic.agent.MaskedMimic attribute) (protomotions.agents.masked_mimic.model.MaskedMimicModel attribute) (protomotions.agents.ppo.agent.PPO attribute) (protomotions.agents.ppo.model.PPOModel attribute) ConfigBuilder (class in protomotions.utils.config_builder) contact_bodies (protomotions.robot_configs.base.RobotConfig attribute) contact_pairs_multiplier (protomotions.robot_configs.base.RobotConfig attribute) (protomotions.robot_configs.smplx.SMPLXRobotConfig attribute) control (protomotions.robot_configs.base.RobotConfig attribute) copy_from() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) copy_from_motion_ids() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) count (protomotions.agents.utils.normalization.RunningMeanStd attribute) create_model() (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.ppo.agent.PPO method) create_optimizers() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.ppo.agent.PPO method) critic (protomotions.agents.ppo.agent.PPO attribute) (protomotions.agents.ppo.config.PPOModelConfig attribute) critic_optimizer (protomotions.agents.ppo.config.PPOModelConfig attribute) critic_step() (protomotions.agents.ppo.agent.PPO method) current_max_episode_length() (protomotions.agents.base_agent.config.MaxEpisodeLengthManagerConfig method)

D

default_dof_pos (protomotions.robot_configs.base.RobotConfig attribute) default_root_height (protomotions.robot_configs.amp.AMPRobotConfig attribute) (protomotions.robot_configs.base.RobotConfig attribute) (protomotions.robot_configs.g1.G1RobotConfig attribute) (protomotions.robot_configs.h1_2.H1_2RobotConfig attribute) (protomotions.robot_configs.smpl.SmplRobotConfig attribute) (protomotions.robot_configs.smplx.SMPLXRobotConfig attribute) device (protomotions.agents.evaluators.base_evaluator.BaseEvaluator property) (protomotions.agents.utils.replay_buffer.ReplayBuffer property) (protomotions.agents.utils.step_tracker.StepTracker property) devices (protomotions.utils.fabric_config.FabricConfig attribute) DictDataset (class in protomotions.agents.utils.data) discount_values() (in module protomotions.agents.ppo.utils) Discriminator (class in protomotions.agents.amp.model) discriminator (protomotions.agents.amp.agent.AMP attribute) (protomotions.agents.amp.config.AMPModelConfig attribute) (protomotions.agents.ase.agent.ASE attribute) discriminator_batch_size (protomotions.agents.amp.config.AMPParametersConfig attribute)

discriminator_grad_penalty (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_logit_weight_decay (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_max_cumulative_bad_transitions (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_optimization_ratio (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_optimizer (protomotions.agents.amp.config.AMPModelConfig attribute) discriminator_replay_keep_prob (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_replay_size (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_reward_threshold (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_reward_w (protomotions.agents.amp.config.AMPParametersConfig attribute) discriminator_step() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method) discriminator_weight_decay (protomotions.agents.amp.config.AMPParametersConfig attribute) DiscriminatorConfig (class in protomotions.agents.amp.config) diversity_bonus (protomotions.agents.ase.config.ASEParametersConfig attribute) diversity_loss() (protomotions.agents.ase.agent.ASE method) diversity_tar (protomotions.agents.ase.config.ASEParametersConfig attribute) done_indices() (protomotions.agents.utils.step_tracker.StepTracker method) dropout (protomotions.agents.common.config.TransformerConfig attribute)

E

e_clip (protomotions.agents.ppo.agent.PPO attribute) (protomotions.agents.ppo.config.PPOAgentConfig attribute) ema_alpha (protomotions.agents.ppo.config.AdvantageNormalizationConfig attribute) enabled (protomotions.agents.ppo.config.AdvantageNormalizationConfig attribute) enc_weights() (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) encoder (protomotions.agents.masked_mimic.config.MaskedMimicModelConfig attribute) encoder_out_size (protomotions.agents.ase.config.ASEDiscriminatorEncoderConfig attribute) end_epoch (protomotions.agents.masked_mimic.config.KLDScheduleConfig attribute) end_kld_coeff (protomotions.agents.masked_mimic.config.KLDScheduleConfig attribute) end_length (protomotions.agents.base_agent.config.MaxEpisodeLengthManagerConfig attribute) end_timer() (protomotions.agents.utils.metering.TimeReport method) env (protomotions.agents.evaluators.base_evaluator.BaseEvaluator property) (protomotions.agents.masked_mimic.agent.MaskedMimic attribute) eps (protomotions.agents.base_agent.config.OptimizerConfig attribute) eval() (protomotions.agents.base_agent.agent.BaseAgent method)

eval_metric_keys (protomotions.agents.evaluators.config.MimicEvaluatorConfig attribute) eval_metrics_every (protomotions.agents.evaluators.config.EvaluatorConfig attribute) evaluate_episode() (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) evaluator (protomotions.agents.base_agent.agent.BaseAgent attribute) (protomotions.agents.base_agent.config.BaseAgentConfig attribute) EvaluatorConfig (class in protomotions.agents.evaluators.config) expand_shape (protomotions.agents.common.config.ModuleOperationExpandConfig attribute) experience_buffer (protomotions.agents.base_agent.agent.BaseAgent attribute) ExperienceBuffer (class in protomotions.agents.utils.data) expert_model (protomotions.agents.masked_mimic.agent.MaskedMimic attribute) expert_model_path (protomotions.agents.masked_mimic.config.MaskedMimicAgentConfig attribute) export_ppo_actor() (in module protomotions.utils.export_utils) export_ppo_critic() (in module protomotions.utils.export_utils) export_ppo_model() (in module protomotions.utils.export_utils)

F

FabricConfig (class in protomotions.utils.fabric_config) fail_on_bad_grads (protomotions.agents.base_agent.config.BaseAgentConfig attribute) FeedForwardModel (class in protomotions.agents.masked_mimic.model) FeedForwardModelConfig (class in protomotions.agents.masked_mimic.config) ff_size (protomotions.agents.common.config.TransformerConfig attribute) fit() (protomotions.agents.base_agent.agent.BaseAgent method) Flatten (class in protomotions.agents.common.common) flatten (protomotions.agents.common.common.Flatten attribute) FlattenConfig (class in protomotions.agents.common.config) forward() (protomotions.agents.amp.model.AMPModel method) (protomotions.agents.amp.model.Discriminator method) (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) (protomotions.agents.base_agent.model.BaseModel method) (protomotions.agents.common.common.Flatten method) (protomotions.agents.common.common.MultiInputModule method) (protomotions.agents.common.common.MultiOutputModule method) (protomotions.agents.common.common.NormObsBase method) (protomotions.agents.common.common.SequentialModule method) (protomotions.agents.common.mlp.MLPWithConcat method) (protomotions.agents.common.transformer.Transformer method) (protomotions.agents.masked_mimic.model.FeedForwardModel method) (protomotions.agents.masked_mimic.model.MaskedMimicModel method) (protomotions.agents.ppo.model.PPOActor method) (protomotions.agents.ppo.model.PPOModel method) (protomotions.utils.export_utils.ONNXExportWrapper method)

from_dict() (protomotions.utils.config_builder.ConfigBuilder class method) from_str() (protomotions.agents.masked_mimic.config.VaeNoiseType class method)

G

G1RobotConfig (class in protomotions.robot_configs.g1) gamma (protomotions.agents.base_agent.config.BaseAgentConfig attribute) GenesisSimulator (built-in class) get() (protomotions.utils.config_builder.ConfigBuilder method) get_activation_func() (in module protomotions.agents.utils.training) get_buffer_size() (protomotions.agents.utils.replay_buffer.ReplayBuffer method) get_class() (in module protomotions.utils.hydra_replacement) get_combined_experience_buffer_rewards() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) get_expert_disc_obs() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method)

get_params() (in module protomotions.agents.common.common) get_state_dict() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.ppo.agent.PPO method) get_step_count_increment() (protomotions.agents.base_agent.agent.BaseAgent method) get_unfilled_mask() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) grad_norm() (in module protomotions.utils.torch_utils) gradient_clip_val (protomotions.agents.base_agent.config.BaseAgentConfig attribute)

H

H1_2RobotConfig (class in protomotions.robot_configs.h1_2)

handle_model_grad_clipping() (in module protomotions.agents.utils.training)

I

import_experiment_relative_eval_overrides() (in module protomotions.utils.config_utils) import_simulator_before_torch() (in module protomotions.utils.simulator_imports) in_keys (protomotions.agents.amp.model.Discriminator attribute) (protomotions.agents.ase.config.ASEDiscriminatorEncoderConfig attribute) (protomotions.agents.base_agent.config.BaseModelConfig attribute) (protomotions.agents.base_agent.model.BaseModel attribute) (protomotions.agents.common.common.Flatten attribute) (protomotions.agents.common.config.FlattenConfig attribute) (protomotions.agents.common.config.MLPWithConcatConfig attribute) (protomotions.agents.common.config.MultiInputModuleConfig attribute) (protomotions.agents.common.config.MultiOutputModuleConfig attribute) (protomotions.agents.common.config.SequentialModuleConfig attribute) (protomotions.agents.common.config.TransformerConfig attribute) (protomotions.agents.common.mlp.MLPWithConcat attribute) (protomotions.agents.common.transformer.Transformer attribute) (protomotions.agents.ppo.config.PPOActorConfig attribute) (protomotions.agents.ppo.model.PPOActor attribute)

init_kld_coeff (protomotions.agents.masked_mimic.config.KLDScheduleConfig attribute) init_state (protomotions.robot_configs.base.RobotConfig attribute) initialize_eval() (protomotions.agents.evaluators.base_evaluator.BaseEvaluator method) (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) input_and_mask_mapping (protomotions.agents.common.config.TransformerConfig attribute) input_models (protomotions.agents.common.config.MultiInputModuleConfig attribute) (protomotions.agents.common.config.SequentialModuleConfig attribute) (protomotions.agents.common.transformer.Transformer attribute) instantiate() (in module protomotions.utils.hydra_replacement) interpolate_pos() (in module protomotions.utils.motion_interpolation_utils) interpolate_quat() (in module protomotions.utils.motion_interpolation_utils) IsaacGymSimulator (built-in class) IsaacLabSimulator (built-in class)

J

jitter_mean_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method)

K

kinematic_info (protomotions.robot_configs.base.RobotConfig attribute) kl_loss() (protomotions.agents.masked_mimic.model.MaskedMimicModel method)

kld_schedule (protomotions.agents.masked_mimic.config.VaeConfig attribute) KLDScheduleConfig (class in protomotions.agents.masked_mimic.config)

L

latent_dim (protomotions.agents.ase.config.ASEParametersConfig attribute) (protomotions.agents.common.config.TransformerConfig attribute) latent_reset_steps (protomotions.agents.ase.agent.ASE attribute) latent_steps_max (protomotions.agents.ase.config.ASEParametersConfig attribute) latent_steps_min (protomotions.agents.ase.config.ASEParametersConfig attribute) latent_uniformity_weight (protomotions.agents.ase.config.ASEParametersConfig attribute) latents (protomotions.agents.ase.agent.ASE attribute) layers (protomotions.agents.common.config.MLPWithConcatConfig attribute) load() (protomotions.agents.base_agent.agent.BaseAgent method)

load_parameters() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.ppo.agent.PPO method) loggers (protomotions.utils.fabric_config.FabricConfig attribute) logit_weights() (protomotions.agents.amp.model.Discriminator method) (protomotions.agents.ase.model.ASEDiscriminatorEncoder method) logstd (protomotions.agents.ppo.model.PPOActor attribute) lr (protomotions.agents.base_agent.config.OptimizerConfig attribute)

M

make_dict() (protomotions.agents.utils.data.ExperienceBuffer method) masked_mimic_step() (protomotions.agents.masked_mimic.agent.MaskedMimic method) MaskedMimic (class in protomotions.agents.masked_mimic.agent) MaskedMimicAgentConfig (class in protomotions.agents.masked_mimic.config) MaskedMimicModel (class in protomotions.agents.masked_mimic.model) MaskedMimicModelConfig (class in protomotions.agents.masked_mimic.config) max_episode_length_manager (protomotions.agents.base_agent.config.BaseAgentConfig attribute) max_eval_steps (protomotions.agents.evaluators.config.MimicEvaluatorConfig attribute) max_mean_reduce() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) max_num_batches() (protomotions.agents.base_agent.agent.BaseAgent method) max_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) MaxEpisodeLengthManagerConfig (class in protomotions.agents.base_agent.config) maybe_clamp() (protomotions.agents.utils.normalization.RunningMeanStd method) mean (protomotions.agents.utils.normalization.RunningMeanStd attribute) mean() (protomotions.agents.utils.metering.TensorAverageMeter method) (protomotions.agents.utils.metering.TensorAverageMeterDict method) mean_and_clear() (protomotions.agents.utils.metering.TensorAverageMeter method) (protomotions.agents.utils.metering.TensorAverageMeterDict method) mean_max_reduce() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) mean_mean_reduce() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) mean_min_reduce() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) mean_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) merge_from() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) mi_enc_forward() (protomotions.agents.ase.agent.ASE method) mi_enc_grad_penalty (protomotions.agents.ase.config.ASEParametersConfig attribute) mi_enc_weight_decay (protomotions.agents.ase.config.ASEParametersConfig attribute) mi_hypersphere_reward_shift (protomotions.agents.ase.config.ASEParametersConfig attribute) mi_reward_w (protomotions.agents.ase.config.ASEParametersConfig attribute) Mimic (built-in class) mimic_small_marker_bodies (protomotions.robot_configs.base.RobotConfig attribute) MimicEvaluator (class in protomotions.agents.evaluators) (class in protomotions.agents.evaluators.mimic_evaluator) MimicEvaluatorConfig (class in protomotions.agents.evaluators.config) min_mean_reduce() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) min_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) min_std (protomotions.agents.ppo.config.AdvantageNormalizationConfig attribute) mlp (protomotions.agents.common.mlp.MLPWithConcat attribute) MLPLayerConfig (class in protomotions.agents.common.config) MLPWithConcat (class in protomotions.agents.common.mlp) MLPWithConcatConfig (class in protomotions.agents.common.config) model (protomotions.agents.amp.config.AMPAgentConfig attribute) (protomotions.agents.ase.agent.ASE attribute) (protomotions.agents.base_agent.agent.BaseAgent attribute) (protomotions.agents.base_agent.config.BaseAgentConfig attribute) (protomotions.agents.masked_mimic.agent.MaskedMimic attribute) (protomotions.agents.masked_mimic.config.MaskedMimicAgentConfig attribute) (protomotions.agents.ppo.config.PPOAgentConfig attribute) module protomotions protomotions.agents protomotions.agents.amp protomotions.agents.amp.agent protomotions.agents.amp.config protomotions.agents.amp.model protomotions.agents.ase protomotions.agents.ase.agent protomotions.agents.ase.config protomotions.agents.ase.model protomotions.agents.base_agent protomotions.agents.base_agent.agent protomotions.agents.base_agent.config protomotions.agents.base_agent.model protomotions.agents.callbacks protomotions.agents.callbacks.slurm_autoresume_srun protomotions.agents.common protomotions.agents.common.common protomotions.agents.common.config protomotions.agents.common.mlp protomotions.agents.common.transformer protomotions.agents.evaluators protomotions.agents.evaluators.base_evaluator protomotions.agents.evaluators.config protomotions.agents.evaluators.metrics protomotions.agents.evaluators.mimic_evaluator protomotions.agents.evaluators.smoothness_evaluator protomotions.agents.masked_mimic protomotions.agents.masked_mimic.agent protomotions.agents.masked_mimic.config protomotions.agents.masked_mimic.model protomotions.agents.mimic protomotions.agents.ppo protomotions.agents.ppo.agent protomotions.agents.ppo.config protomotions.agents.ppo.model protomotions.agents.ppo.utils protomotions.agents.utils protomotions.agents.utils.data protomotions.agents.utils.metering protomotions.agents.utils.normalization protomotions.agents.utils.replay_buffer protomotions.agents.utils.step_tracker protomotions.agents.utils.training protomotions.tests protomotions.utils protomotions.utils.component_builder protomotions.utils.config_builder protomotions.utils.config_utils protomotions.utils.export_utils protomotions.utils.fabric_config protomotions.utils.hydra_replacement protomotions.utils.inference_utils protomotions.utils.mesh_utils protomotions.utils.motion_interpolation_utils protomotions.utils.rotations protomotions.utils.simulator_imports protomotions.utils.torch_utils

module_operations (protomotions.agents.common.config.FlattenConfig attribute) (protomotions.agents.common.config.MLPWithConcatConfig attribute) ModuleOperationConfig (class in protomotions.agents.common.config) ModuleOperationExpandConfig (class in protomotions.agents.common.config) ModuleOperationForwardConfig (class in protomotions.agents.common.config) ModuleOperationPermuteConfig (class in protomotions.agents.common.config) ModuleOperationReshapeConfig (class in protomotions.agents.common.config) ModuleOperationSphereProjectionConfig (class in protomotions.agents.common.config) ModuleOperationSqueezeConfig (class in protomotions.agents.common.config) ModuleOperationUnsqueezeConfig (class in protomotions.agents.common.config) motion_lib (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator property) (protomotions.agents.evaluators.MimicEvaluator property) motion_manager (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator property) (protomotions.agents.evaluators.MimicEvaluator property) MotionMetrics (class in protomotions.agents.evaluators) (class in protomotions.agents.evaluators.metrics) mu (protomotions.agents.ppo.model.PPOActor attribute) mu_key (protomotions.agents.ppo.config.PPOActorConfig attribute) mu_model (protomotions.agents.ppo.config.PPOActorConfig attribute) MultiInputModule (class in protomotions.agents.common.common) MultiInputModuleConfig (class in protomotions.agents.common.config) MultiOutputModule (class in protomotions.agents.common.common) MultiOutputModuleConfig (class in protomotions.agents.common.config)

N

new_order (protomotions.agents.common.config.ModuleOperationPermuteConfig attribute) new_shape (protomotions.agents.common.config.ModuleOperationReshapeConfig attribute) NewtonSimulator (built-in class) non_termination_contact_bodies (protomotions.robot_configs.base.RobotConfig attribute) norm (protomotions.agents.common.common.Flatten attribute) (protomotions.agents.common.mlp.MLPWithConcat attribute) norm_clamp_value (protomotions.agents.common.config.NormObsBaseConfig attribute) NORMAL (protomotions.agents.masked_mimic.config.VaeNoiseType attribute) normalize() (protomotions.agents.utils.normalization.RewardRunningMeanStd method) (protomotions.agents.utils.normalization.RunningMeanStd method) normalize_obs (protomotions.agents.common.config.NormObsBaseConfig attribute) normalize_rewards (protomotions.agents.base_agent.config.BaseAgentConfig attribute) normalized_reward_clamp_value (protomotions.agents.base_agent.config.BaseAgentConfig attribute) NormObsBase (class in protomotions.agents.common.common)

NormObsBaseConfig (class in protomotions.agents.common.config) num_batches() (protomotions.agents.utils.data.DictDataset method) num_envs (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator property) (protomotions.agents.evaluators.MimicEvaluator property) num_heads (protomotions.agents.common.config.TransformerConfig attribute) num_layers (protomotions.agents.common.config.TransformerConfig attribute) num_mini_epochs (protomotions.agents.base_agent.config.BaseAgentConfig attribute) num_nodes (protomotions.utils.fabric_config.FabricConfig attribute) num_ons (protomotions.agents.utils.metering.Timer attribute) num_out (protomotions.agents.common.common.NormObsBase attribute) (protomotions.agents.common.config.MLPWithConcatConfig attribute) (protomotions.agents.ppo.config.PPOActorConfig attribute) num_steps (protomotions.agents.base_agent.config.BaseAgentConfig attribute) number_of_actions (protomotions.robot_configs.base.RobotConfig attribute)

O

obs_dict_to_tensordict() (protomotions.agents.base_agent.agent.BaseAgent method) off() (protomotions.agents.utils.metering.Timer method) on() (protomotions.agents.utils.metering.Timer method) on_fit_end() (protomotions.agents.callbacks.slurm_autoresume_srun.AutoResumeCallbackSrun method) on_fit_start() (protomotions.agents.callbacks.slurm_autoresume_srun.AutoResumeCallbackSrun method) ONNXExportWrapper (class in protomotions.utils.export_utils) ops_mean_reduce() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) optimize_model() (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.ppo.agent.PPO method) optimizer (protomotions.agents.base_agent.agent.BaseAgent attribute) (protomotions.agents.masked_mimic.config.MaskedMimicModelConfig attribute) OptimizerConfig (class in protomotions.agents.base_agent.config) out_keys (protomotions.agents.amp.config.DiscriminatorConfig attribute) (protomotions.agents.amp.model.Discriminator attribute) (protomotions.agents.ase.config.ASEDiscriminatorEncoderConfig attribute) (protomotions.agents.base_agent.config.BaseModelConfig attribute) (protomotions.agents.base_agent.model.BaseModel attribute) (protomotions.agents.common.common.Flatten attribute) (protomotions.agents.common.config.FlattenConfig attribute) (protomotions.agents.common.config.MLPWithConcatConfig attribute) (protomotions.agents.common.config.MultiInputModuleConfig attribute) (protomotions.agents.common.config.MultiOutputModuleConfig attribute) (protomotions.agents.common.config.SequentialModuleConfig attribute) (protomotions.agents.common.config.TransformerConfig attribute) (protomotions.agents.common.mlp.MLPWithConcat attribute) (protomotions.agents.common.transformer.Transformer attribute) (protomotions.agents.ppo.config.PPOActorConfig attribute) (protomotions.agents.ppo.config.PPOModelConfig attribute) (protomotions.agents.ppo.model.PPOActor attribute)

output_activation (protomotions.agents.common.config.MLPWithConcatConfig attribute) (protomotions.agents.common.config.TransformerConfig attribute) output_models (protomotions.agents.common.config.MultiOutputModuleConfig attribute)

P

parse_cli_overrides() (in module protomotions.utils.config_utils) PathFollowing (built-in class) perform_optimization_step() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.ppo.agent.PPO method) pop_timer() (protomotions.agents.utils.metering.TimeReport method) post_env_step_modifications() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) post_epoch_logging() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) PPO (class in protomotions.agents.ppo.agent) PPOActor (class in protomotions.agents.ppo.model) PPOActorConfig (class in protomotions.agents.ppo.config) PPOAgentConfig (class in protomotions.agents.ppo.config) PPOModel (class in protomotions.agents.ppo.model) PPOModelConfig (class in protomotions.agents.ppo.config) precision (protomotions.utils.fabric_config.FabricConfig attribute) prior (protomotions.agents.masked_mimic.config.MaskedMimicModelConfig attribute) process_eval_results() (protomotions.agents.evaluators.base_evaluator.BaseEvaluator method) (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) produce_negative_expert_obs() (protomotions.agents.ase.agent.ASE method) protomotions module protomotions.agents module protomotions.agents.amp module protomotions.agents.amp.agent module protomotions.agents.amp.config module protomotions.agents.amp.model module protomotions.agents.ase module protomotions.agents.ase.agent module protomotions.agents.ase.config module protomotions.agents.ase.model module protomotions.agents.base_agent module protomotions.agents.base_agent.agent module protomotions.agents.base_agent.config module protomotions.agents.base_agent.model module protomotions.agents.callbacks module protomotions.agents.callbacks.slurm_autoresume_srun module protomotions.agents.common module protomotions.agents.common.common module protomotions.agents.common.config module protomotions.agents.common.mlp module protomotions.agents.common.transformer module protomotions.agents.evaluators module protomotions.agents.evaluators.base_evaluator module

protomotions.agents.evaluators.config module protomotions.agents.evaluators.metrics module protomotions.agents.evaluators.mimic_evaluator module protomotions.agents.evaluators.smoothness_evaluator module protomotions.agents.masked_mimic module protomotions.agents.masked_mimic.agent module protomotions.agents.masked_mimic.config module protomotions.agents.masked_mimic.model module protomotions.agents.mimic module protomotions.agents.ppo module protomotions.agents.ppo.agent module protomotions.agents.ppo.config module protomotions.agents.ppo.model module protomotions.agents.ppo.utils module protomotions.agents.utils module protomotions.agents.utils.data module protomotions.agents.utils.metering module protomotions.agents.utils.normalization module protomotions.agents.utils.replay_buffer module protomotions.agents.utils.step_tracker module protomotions.agents.utils.training module protomotions.tests module protomotions.utils module protomotions.utils.component_builder module protomotions.utils.config_builder module protomotions.utils.config_utils module protomotions.utils.export_utils module protomotions.utils.fabric_config module protomotions.utils.hydra_replacement module protomotions.utils.inference_utils module protomotions.utils.mesh_utils module protomotions.utils.motion_interpolation_utils module protomotions.utils.rotations module protomotions.utils.simulator_imports module protomotions.utils.torch_utils module

R

record_reward() (protomotions.agents.utils.normalization.RewardRunningMeanStd method) record_rollout_step() (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.ppo.agent.PPO method) register_algorithm_experience_buffer_keys() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) (protomotions.agents.ppo.agent.PPO method) register_key() (protomotions.agents.utils.data.ExperienceBuffer method) reparameterization() (protomotions.agents.masked_mimic.model.MaskedMimicModel static method) ReplayBuffer (class in protomotions.agents.utils.replay_buffer) report() (protomotions.agents.utils.metering.Timer method) (protomotions.agents.utils.metering.TimeReport method) reset() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) (protomotions.agents.utils.replay_buffer.ReplayBuffer method)

reset_latents() (protomotions.agents.ase.agent.ASE method) reset_steps() (protomotions.agents.utils.step_tracker.StepTracker method) reset_vae_noise() (protomotions.agents.masked_mimic.agent.MaskedMimic method) RewardRunningMeanStd (class in protomotions.agents.utils.normalization) RobotConfig (class in protomotions.robot_configs.base) root_dir (protomotions.agents.evaluators.base_evaluator.BaseEvaluator property) rotation_jitter_mean_reduce_each_motion() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) run_evaluation() (protomotions.agents.evaluators.base_evaluator.BaseEvaluator method) (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) running_obs_norm (protomotions.agents.common.common.NormObsBase attribute) RunningMeanStd (class in protomotions.agents.utils.normalization)

S

sample() (protomotions.agents.utils.replay_buffer.ReplayBuffer method) sample_latents() (protomotions.agents.ase.agent.ASE method) save() (protomotions.agents.base_agent.agent.BaseAgent method) save_epoch_checkpoint_every (protomotions.agents.base_agent.config.BaseAgentConfig attribute) save_last_checkpoint_every (protomotions.agents.base_agent.config.BaseAgentConfig attribute) save_predicted_motion_lib_every (protomotions.agents.evaluators.config.MimicEvaluatorConfig attribute) seeding() (in module protomotions.utils.torch_utils) seqTransEncoder (protomotions.agents.common.transformer.Transformer attribute) sequence_pos_encoder (protomotions.agents.common.transformer.Transformer attribute) sequential_models (protomotions.agents.amp.model.Discriminator attribute) SequentialModule (class in protomotions.agents.common.common) SequentialModuleConfig (class in protomotions.agents.common.config) setup() (protomotions.agents.ase.agent.ASE method) (protomotions.agents.base_agent.agent.BaseAgent method) (protomotions.agents.masked_mimic.agent.MaskedMimic method) shift_counter() (protomotions.agents.utils.step_tracker.StepTracker method) shift_mean (protomotions.agents.ppo.config.AdvantageNormalizationConfig attribute) should_stop (protomotions.agents.base_agent.agent.BaseAgent property) shuffle() (protomotions.agents.utils.data.DictDataset method) simple_test_policy() (protomotions.agents.evaluators.base_evaluator.BaseEvaluator method) (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method)

simulation_params (protomotions.robot_configs.amp.AMPRobotConfig attribute) (protomotions.robot_configs.base.RobotConfig attribute) (protomotions.robot_configs.g1.G1RobotConfig attribute) (protomotions.robot_configs.h1_2.H1_2RobotConfig attribute) (protomotions.robot_configs.smpl.SmplRobotConfig attribute) (protomotions.robot_configs.smplx.SMPLXRobotConfig attribute) Simulator (built-in class) SmoothnessEvaluator (class in protomotions.agents.evaluators) (class in protomotions.agents.evaluators.smoothness_evaluator) SmoothnessMetricPlugin (class in protomotions.agents.evaluators.base_evaluator) SmplRobotConfig (class in protomotions.robot_configs.smpl) SMPLXRobotConfig (class in protomotions.robot_configs.smplx) squeeze_dim (protomotions.agents.common.config.ModuleOperationSqueezeConfig attribute) start_epoch (protomotions.agents.masked_mimic.config.KLDScheduleConfig attribute) start_length (protomotions.agents.base_agent.config.MaxEpisodeLengthManagerConfig attribute) start_timer() (protomotions.agents.utils.metering.TimeReport method) Steering (built-in class) steps (protomotions.agents.utils.step_tracker.StepTracker attribute) StepTracker (class in protomotions.agents.utils.step_tracker) store() (protomotions.agents.utils.replay_buffer.ReplayBuffer method) store_dict (protomotions.agents.utils.data.ExperienceBuffer attribute) store_latents() (protomotions.agents.ase.agent.ASE method) swap_and_flatten01() (in module protomotions.agents.utils.data)

T

task_reward_w (protomotions.agents.base_agent.config.BaseAgentConfig attribute) tau (protomotions.agents.ppo.agent.PPO attribute) (protomotions.agents.ppo.config.PPOAgentConfig attribute) TensorAverageMeter (class in protomotions.agents.utils.metering) TensorAverageMeterDict (class in protomotions.agents.utils.metering) terminate_early() (protomotions.agents.amp.agent.AMP method) (protomotions.agents.base_agent.agent.BaseAgent method) test_implementations_comparison() (protomotions.utils.rotations.TestQuatAngleAxis method) test_orthogonal_vectors() (protomotions.utils.rotations.TestQuatFromTwoVectors method) test_same_vectors() (protomotions.utils.rotations.TestQuatFromTwoVectors method) test_w_first_format() (protomotions.utils.rotations.TestQuatFromTwoVectors method) TestQuatAngleAxis (class in protomotions.utils.rotations) TestQuatFromTwoVectors (class in protomotions.utils.rotations) time_total (protomotions.agents.utils.metering.Timer attribute) Timer (class in protomotions.agents.utils.metering) TimeReport (class in protomotions.agents.utils.metering)

to() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) (protomotions.agents.utils.normalization.RunningMeanStd method) to_dict() (protomotions.utils.config_builder.ConfigBuilder method) to_torch() (in module protomotions.utils.torch_utils) total_sum() (protomotions.agents.utils.data.ExperienceBuffer method) trackable_bodies_subset (protomotions.robot_configs.base.RobotConfig attribute) train() (protomotions.agents.base_agent.agent.BaseAgent method) training_early_termination (protomotions.agents.base_agent.config.BaseAgentConfig attribute) training_max_steps (protomotions.agents.base_agent.config.BaseAgentConfig attribute) Transformer (class in protomotions.agents.common.transformer) transformer_token_size (protomotions.agents.common.config.TransformerConfig attribute) TransformerConfig (class in protomotions.agents.common.config) transition_epochs (protomotions.agents.base_agent.config.MaxEpisodeLengthManagerConfig attribute) trunk (protomotions.agents.masked_mimic.config.FeedForwardModelConfig attribute) (protomotions.agents.masked_mimic.config.MaskedMimicModelConfig attribute)

U

UNIFORM (protomotions.agents.masked_mimic.config.VaeNoiseType attribute) uniformity_kernel_scale (protomotions.agents.ase.config.ASEParametersConfig attribute) units (protomotions.agents.common.config.MLPLayerConfig attribute) unsqueeze_dim (protomotions.agents.common.config.ModuleOperationUnsqueezeConfig attribute) update() (protomotions.agents.evaluators.metrics.MotionMetrics method) (protomotions.agents.evaluators.MotionMetrics method) update_data() (protomotions.agents.utils.data.ExperienceBuffer method)

update_disc_replay_buffer() (protomotions.agents.amp.agent.AMP method) update_fields() (protomotions.robot_configs.base.RobotConfig method) update_latents() (protomotions.agents.ase.agent.ASE method) update_metrics_from_env_extras() (protomotions.agents.evaluators.mimic_evaluator.MimicEvaluator method) (protomotions.agents.evaluators.MimicEvaluator method) use_ema (protomotions.agents.ppo.config.AdvantageNormalizationConfig attribute) use_layer_norm (protomotions.agents.common.config.MLPLayerConfig attribute)

V

vae (protomotions.agents.masked_mimic.config.MaskedMimicModelConfig attribute) vae_latent_dim (protomotions.agents.masked_mimic.config.VaeConfig attribute) vae_noise (protomotions.agents.masked_mimic.agent.MaskedMimic attribute)

vae_noise_type (protomotions.agents.masked_mimic.config.VaeConfig attribute) VaeConfig (class in protomotions.agents.masked_mimic.config) VaeNoiseType (class in protomotions.agents.masked_mimic.config) var (protomotions.agents.utils.normalization.RunningMeanStd attribute)

W

wandb_run_exists() (in module protomotions.agents.callbacks.slurm_autoresume_srun)

weight_decay (protomotions.agents.base_agent.config.OptimizerConfig attribute) weight_init() (in module protomotions.agents.common.common)

Z

ZEROS (protomotions.agents.masked_mimic.config.VaeNoiseType attribute)