fishyrl package

Submodules

• fishyrl.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()
• fishyrl.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()
• fishyrl.distributions module
  • TwoHot
    • TwoHot.__init__()
    • TwoHot.entropy()
    • TwoHot.log_prob()
    • TwoHot.rsample()
    • TwoHot.mean
  • identity()
  • symexp()
  • symlog()
  • uniform_mix()
• fishyrl.dreamer module
  • compute_actions()
  • construct_models()
  • evaluate()
  • get_environments_and_actions()
  • learning_step()
  • load_models()
  • save_models()
  • train_loop()
• fishyrl.losses module
  • mse_loss()
• fishyrl.models module
  • Actor
    • Actor.__init__()
    • Actor.forward()
  • CNNDecoder
    • CNNDecoder.__init__()
    • CNNDecoder.forward()
  • CNNEncoder
    • CNNEncoder.__init__()
    • CNNEncoder.forward()
    • CNNEncoder.output_dim
  • ChannelNorm
    • ChannelNorm.__init__()
    • ChannelNorm.forward()
  • LayerNormGRU
    • LayerNormGRU.__init__()
    • LayerNormGRU.forward()
  • MLP
    • MLP.__init__()
    • MLP.forward()
  • MLPDecoder
    • MLPDecoder.__init__()
    • MLPDecoder.forward()
  • MLPEncoder
    • MLPEncoder.__init__()
    • MLPEncoder.forward()
  • RSSM
    • RSSM.__init__()
    • RSSM.forward()
    • RSSM.infer_stochastic()
    • RSSM.initial_hidden_state
  • RecurrentModel
    • RecurrentModel.__init__()
    • RecurrentModel.forward()
    • RecurrentModel.hidden_dim
• fishyrl.utilities module
  • 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()
  • init_weights()
  • load_config()
  • optional_flatten_cfg()
  • uniform_init_weights()

Module contents