dreamerx package

Subpackages

• dreamerx.rlgym package
  • Submodules
    • dreamerx.rlgym.rewards module
      • CloseReward
  • Module contents

Submodules

• dreamerx.actions module
  • ACTION_IDENTIFIERS
    • ACTION_IDENTIFIERS.CONTINUOUS
    • ACTION_IDENTIFIERS.DISCRETE
    • ACTION_IDENTIFIERS.DISCRETIZED_CONTINUOUS
    • ACTION_IDENTIFIERS.TWO_HOT_DISCRETIZED_CONTINUOUS
  • Action
    • Action.construct()
    • Action.sample()
    • Action.simplify()
    • Action.input_dim
    • Action.num_actions
    • Action.output_dim
  • ContinuousActions
    • ContinuousActions.__init__()
    • ContinuousActions.construct()
    • ContinuousActions.sample()
    • ContinuousActions.simplify()
    • ContinuousActions.input_dim
    • ContinuousActions.num_actions
    • ContinuousActions.output_dim
  • DiscreteAction
    • DiscreteAction.__init__()
    • DiscreteAction.construct()
    • DiscreteAction.sample()
    • DiscreteAction.simplify()
    • DiscreteAction.input_dim
    • DiscreteAction.num_actions
    • DiscreteAction.output_dim
  • DiscretizedContinuousAction
    • DiscretizedContinuousAction.__init__()
    • DiscretizedContinuousAction.construct()
    • DiscretizedContinuousAction.sample()
    • DiscretizedContinuousAction.simplify()
    • DiscretizedContinuousAction.input_dim
    • DiscretizedContinuousAction.num_actions
    • DiscretizedContinuousAction.output_dim
  • TwoHotDiscretizedContinuousAction
    • TwoHotDiscretizedContinuousAction.__init__()
    • TwoHotDiscretizedContinuousAction.construct()
    • TwoHotDiscretizedContinuousAction.sample()
    • TwoHotDiscretizedContinuousAction.simplify()
    • TwoHotDiscretizedContinuousAction.input_dim
    • TwoHotDiscretizedContinuousAction.num_actions
    • TwoHotDiscretizedContinuousAction.output_dim
  • construct_actions()
  • simplify_actions()
• dreamerx.buffers module
  • Buffer
    • Buffer.add()
    • Buffer.load_state_dict()
    • Buffer.reset()
    • Buffer.sample()
    • Buffer.state_dict()
    • Buffer.size
  • IndependentVectorizedBuffer
    • IndependentVectorizedBuffer.__init__()
    • IndependentVectorizedBuffer.add()
    • IndependentVectorizedBuffer.load_state_dict()
    • IndependentVectorizedBuffer.reset()
    • IndependentVectorizedBuffer.sample()
    • IndependentVectorizedBuffer.state_dict()
    • IndependentVectorizedBuffer.size
  • SequentialBuffer
    • SequentialBuffer.__init__()
    • SequentialBuffer.add()
    • SequentialBuffer.load_state_dict()
    • SequentialBuffer.reset()
    • SequentialBuffer.sample()
    • SequentialBuffer.state_dict()
    • SequentialBuffer.capacity
    • SequentialBuffer.is_full
    • SequentialBuffer.size
  • convert_samples_to_tensors()
• dreamerx.distributions module
  • TwoHot
    • TwoHot.__init__()
    • TwoHot.entropy()
    • TwoHot.log_prob()
    • TwoHot.rsample()
    • TwoHot.mean
    • TwoHot.mode
  • identity()
  • symexp()
  • symlog()
  • uniform_mix()
• dreamerx.dreamer module
  • compute_actions()
  • construct_actions()
  • construct_envs()
  • construct_models()
  • evaluate()
  • learning_step()
  • load_models()
  • save_models()
  • train_loop()
• dreamerx.environments module
  • VectorizedEnvironment
    • VectorizedEnvironment.action_sample()
    • VectorizedEnvironment.copy()
    • VectorizedEnvironment.render()
    • VectorizedEnvironment.reset()
    • VectorizedEnvironment.step()
    • VectorizedEnvironment.num_envs
    • VectorizedEnvironment.obs_shape
    • VectorizedEnvironment.render_fps
  • VectorizedGymEnvironment
    • VectorizedGymEnvironment.__init__()
    • VectorizedGymEnvironment.action_sample()
    • VectorizedGymEnvironment.copy()
    • VectorizedGymEnvironment.render()
    • VectorizedGymEnvironment.reset()
    • VectorizedGymEnvironment.step()
    • VectorizedGymEnvironment.num_envs
    • VectorizedGymEnvironment.obs_shape
    • VectorizedGymEnvironment.render_fps
  • VectorizedRLGymEnvironment
    • VectorizedRLGymEnvironment.__init__()
    • VectorizedRLGymEnvironment._construct_actions()
    • VectorizedRLGymEnvironment.action_sample()
    • VectorizedRLGymEnvironment.copy()
    • VectorizedRLGymEnvironment.render()
    • VectorizedRLGymEnvironment.reset()
    • VectorizedRLGymEnvironment.step()
    • VectorizedRLGymEnvironment.num_envs
    • VectorizedRLGymEnvironment.obs_shape
    • VectorizedRLGymEnvironment.render_fps
  • register_env()
• dreamerx.losses module
  • attention_reconstruction_loss()
  • hungarian_loss()
  • mse_loss()
• dreamerx.models module
  • Actor
    • Actor.__init__()
    • Actor.forward()
  • AttentionDecoder
    • AttentionDecoder.__init__()
    • AttentionDecoder.forward()
  • AttentionEncoder
    • AttentionEncoder.__init__()
    • AttentionEncoder.forward()
    • AttentionEncoder.output_dim
  • BlockLinear
    • BlockLinear.__init__()
    • BlockLinear.forward()
  • BlockRecurrentModel
    • BlockRecurrentModel.__init__()
    • BlockRecurrentModel.forward()
    • BlockRecurrentModel.deter_dim
  • CNNDecoder
    • CNNDecoder.__init__()
    • CNNDecoder.forward()
  • CNNEncoder
    • CNNEncoder.__init__()
    • CNNEncoder.forward()
    • CNNEncoder.output_dim
  • ChannelNorm
    • ChannelNorm.__init__()
    • ChannelNorm.forward()
  • CompoundDecoder
    • CompoundDecoder.__init__()
    • CompoundDecoder.forward()
  • CompoundEncoder
    • CompoundEncoder.__init__()
    • CompoundEncoder.forward()
    • CompoundEncoder.output_dim
  • LayerNormGRU
    • LayerNormGRU.__init__()
    • LayerNormGRU.forward()
  • MLP
    • MLP.__init__()
    • MLP.forward()
  • MLPDecoder
    • MLPDecoder.__init__()
    • MLPDecoder.forward()
  • MLPEncoder
    • MLPEncoder.__init__()
    • MLPEncoder.forward()
    • MLPEncoder.output_dim
  • PositionalEncoding
    • PositionalEncoding.__init__()
    • PositionalEncoding.forward()
  • RSSM
    • RSSM.__init__()
    • RSSM.forward()
    • RSSM.infer_stochastic()
    • RSSM.initial_hidden_state
  • SingleRecurrentModel
    • SingleRecurrentModel.__init__()
    • SingleRecurrentModel.forward()
    • SingleRecurrentModel.deter_dim
  • extract_representation()
• dreamerx.utilities module
  • CaseInsensitiveDict
  • CaseInsensitiveEnumMeta
  • Container
    • Container.__init__()
    • Container.load_state_dict()
    • Container.state_dict()
  • ContainerModule
    • ContainerModule.__init__()
  • DotDict
    • DotDict.__init__()
    • DotDict._crawl()
  • MovingMinMaxScaler
    • MovingMinMaxScaler.__init__()
    • MovingMinMaxScaler.forward()
  • Ratio
    • Ratio.__init__()
    • Ratio.load_state_dict()
    • Ratio.state_dict()
  • _flatten_dict()
  • _merge_dotdicts()
  • export_frames()
  • init_weights()
  • load_config()
  • optional_flatten_cfg()
  • uniform_init_weights()

Module contents