protomotions.agents.common package

Submodules

• protomotions.agents.common.common module
  • get_params()
  • weight_init()
  • NormObsBase
    • running_obs_norm
    • num_out
    • __init__()
    • build_norm()
    • forward()
  • apply_module_operations()
  • Flatten
    • norm
    • flatten
    • in_keys
    • out_keys
    • __init__()
    • config
    • forward()
  • SequentialModule
    • __init__()
    • config
    • forward()
  • MultiInputModule
    • __init__()
    • config
    • forward()
  • MultiOutputModule
    • __init__()
    • config
    • forward()
• protomotions.agents.common.config module
  • NormObsBaseConfig
    • normalize_obs
    • norm_clamp_value
    • __init__()
  • ModuleOperationConfig
    • __init__()
  • ModuleOperationForwardConfig
    • __init__()
  • ModuleOperationPermuteConfig
    • new_order
    • __init__()
  • ModuleOperationReshapeConfig
    • new_shape
    • __init__()
  • ModuleOperationSqueezeConfig
    • squeeze_dim
    • __init__()
  • ModuleOperationUnsqueezeConfig
    • unsqueeze_dim
    • __init__()
  • ModuleOperationExpandConfig
    • expand_shape
    • __init__()
  • ModuleOperationSphereProjectionConfig
    • __init__()
  • FlattenConfig
    • in_keys
    • out_keys
    • module_operations
    • __init__()
  • MLPLayerConfig
    • units
    • activation
    • use_layer_norm
    • __init__()
  • MLPWithConcatConfig
    • num_out
    • layers
    • in_keys
    • out_keys
    • output_activation
    • module_operations
    • __init__()
  • MultiInputModuleConfig
    • input_models
    • in_keys
    • out_keys
    • __init__()
  • SequentialModuleConfig
    • input_models
    • in_keys
    • out_keys
    • __init__()
  • MultiOutputModuleConfig
    • output_models
    • in_keys
    • out_keys
    • __init__()
  • TransformerConfig
    • in_keys
    • out_keys
    • input_and_mask_mapping
    • transformer_token_size
    • latent_dim
    • num_heads
    • ff_size
    • num_layers
    • dropout
    • activation
    • output_activation
    • __init__()
• protomotions.agents.common.mlp module
  • build_mlp()
  • MLPWithConcat
    • norm
    • mlp
    • in_keys
    • out_keys
    • __init__()
    • config
    • forward()
• protomotions.agents.common.transformer module
  • Transformer
    • input_models
    • sequence_pos_encoder
    • seqTransEncoder
    • in_keys
    • out_keys
    • __init__()
    • forward()