protomotions.agents package

Subpackages

• protomotions.agents.amp package
  • Submodules
    • protomotions.agents.amp.agent module
      • AMP
    • protomotions.agents.amp.config module
      • AMPParametersConfig
      • DiscriminatorConfig
      • AMPModelConfig
      • AMPAgentConfig
    • protomotions.agents.amp.model module
      • Discriminator
      • AMPModel
• protomotions.agents.ase package
  • Submodules
    • protomotions.agents.ase.agent module
      • ASE
    • protomotions.agents.ase.config module
      • ASEParametersConfig
      • ASEDiscriminatorEncoderConfig
      • ASEAgentConfig
    • protomotions.agents.ase.model module
      • ASEDiscriminatorEncoder
• protomotions.agents.base_agent package
  • Submodules
    • protomotions.agents.base_agent.agent module
      • BaseAgent
    • protomotions.agents.base_agent.config module
      • MaxEpisodeLengthManagerConfig
      • OptimizerConfig
      • BaseModelConfig
      • BaseAgentConfig
    • protomotions.agents.base_agent.model module
      • BaseModel
• protomotions.agents.callbacks package
  • Submodules
    • protomotions.agents.callbacks.slurm_autoresume_srun module
      • wandb_run_exists()
      • AutoResumeCallbackSrun
• protomotions.agents.common package
  • Submodules
    • protomotions.agents.common.common module
      • get_params()
      • weight_init()
      • NormObsBase
      • apply_module_operations()
      • Flatten
      • SequentialModule
      • MultiInputModule
      • MultiOutputModule
    • protomotions.agents.common.config module
      • NormObsBaseConfig
      • ModuleOperationConfig
      • ModuleOperationForwardConfig
      • ModuleOperationPermuteConfig
      • ModuleOperationReshapeConfig
      • ModuleOperationSqueezeConfig
      • ModuleOperationUnsqueezeConfig
      • ModuleOperationExpandConfig
      • ModuleOperationSphereProjectionConfig
      • FlattenConfig
      • MLPLayerConfig
      • MLPWithConcatConfig
      • MultiInputModuleConfig
      • SequentialModuleConfig
      • MultiOutputModuleConfig
      • TransformerConfig
    • protomotions.agents.common.mlp module
      • build_mlp()
      • MLPWithConcat
    • protomotions.agents.common.transformer module
      • Transformer
• protomotions.agents.evaluators package
  • MimicEvaluator
    • __init__()
    • add_extra_obs_to_agent()
    • cleanup_after_evaluation()
    • evaluate_episode()
    • initialize_eval()
    • motion_lib
    • motion_manager
    • num_envs
    • process_eval_results()
    • run_evaluation()
    • simple_test_policy()
    • update_metrics_from_env_extras()
  • SmoothnessEvaluator
    • __init__()
    • compute_normalized_jerk_from_pos()
    • compute_smoothness_metrics()
  • MotionMetrics
    • __init__()
    • compute_finite_difference_jitter_reduce_each_motion()
    • compute_jitter_reduce_each_motion()
    • compute_rotation_jitter_reduce_each_motion()
    • copy_from()
    • copy_from_motion_ids()
    • get_unfilled_mask()
    • jitter_mean_reduce_each_motion()
    • max_mean_reduce()
    • max_reduce_each_motion()
    • mean_max_reduce()
    • mean_mean_reduce()
    • mean_min_reduce()
    • mean_reduce_each_motion()
    • merge_from()
    • min_mean_reduce()
    • min_reduce_each_motion()
    • ops_mean_reduce()
    • reset()
    • rotation_jitter_mean_reduce_each_motion()
    • to()
    • update()
  • Submodules
    • protomotions.agents.evaluators.base_evaluator module
      • SmoothnessMetricPlugin
      • BaseEvaluator
    • protomotions.agents.evaluators.config module
      • EvaluatorConfig
      • MimicEvaluatorConfig
    • protomotions.agents.evaluators.metrics module
      • MotionMetrics
    • protomotions.agents.evaluators.mimic_evaluator module
      • MimicEvaluator
    • protomotions.agents.evaluators.smoothness_evaluator module
      • SmoothnessEvaluator
• protomotions.agents.masked_mimic package
  • Submodules
    • protomotions.agents.masked_mimic.agent module
      • MaskedMimic
    • protomotions.agents.masked_mimic.config module
      • KLDScheduleConfig
      • VaeNoiseType
      • VaeConfig
      • FeedForwardModelConfig
      • MaskedMimicModelConfig
      • MaskedMimicAgentConfig
    • protomotions.agents.masked_mimic.model module
      • FeedForwardModel
      • MaskedMimicModel
• protomotions.agents.mimic package
  • Submodules
• protomotions.agents.ppo package
  • Submodules
    • protomotions.agents.ppo.agent module
      • PPO
    • protomotions.agents.ppo.config module
      • PPOActorConfig
      • PPOModelConfig
      • AdvantageNormalizationConfig
      • PPOAgentConfig
    • protomotions.agents.ppo.model module
      • PPOActor
      • PPOModel
    • protomotions.agents.ppo.utils module
      • discount_values()
• protomotions.agents.utils package
  • Submodules
    • protomotions.agents.utils.data module
      • swap_and_flatten01()
      • ExperienceBuffer
      • DictDataset
    • protomotions.agents.utils.metering module
      • Timer
      • TimeReport
      • TensorAverageMeter
      • TensorAverageMeterDict
    • protomotions.agents.utils.normalization module
      • RunningMeanStd
      • combine_moments()
      • RewardRunningMeanStd
    • protomotions.agents.utils.replay_buffer module
      • ReplayBuffer
    • protomotions.agents.utils.step_tracker module
      • StepTracker
    • protomotions.agents.utils.training module
      • bounds_loss()
      • handle_model_grad_clipping()
      • aggregate_scalar_metrics()
      • get_activation_func()