src package

Subpackages

• src.agents package
  • Subpackages
    • src.agents.petsplanners package
      • Submodules
      • src.agents.petsplanners.CEMPlanner module
      • src.agents.petsplanners.PDDMPlanner module
      • src.agents.petsplanners.RandomPlanner module
      • src.agents.petsplanners.base module
      • Module contents
  • Submodules
  • src.agents.PETSAgent module
    • PETSAgent
      • PETSAgent.instantiate_from_config()
      • PETSAgent.instantiate_from_config_dict()
      • PETSAgent.load_model()
      • PETSAgent.register_reset()
      • PETSAgent.save_model()
      • PETSAgent.schema
      • PETSAgent.select_action()
      • PETSAgent.update()
  • src.agents.PPOAgent module
    • PPOAgent
      • PPOAgent.instantiate_from_config()
      • PPOAgent.instantiate_from_config_dict()
      • PPOAgent.load_model()
      • PPOAgent.register_reset()
      • PPOAgent.save_model()
      • PPOAgent.schema
      • PPOAgent.select_action()
      • PPOAgent.update()
  • src.agents.SACAgent module
    • SACAgent
      • SACAgent.instantiate_from_config()
      • SACAgent.instantiate_from_config_dict()
      • SACAgent.load_model()
      • SACAgent.register_reset()
      • SACAgent.save_model()
      • SACAgent.schema
      • SACAgent.select_action()
      • SACAgent.update()
  • src.agents.base module
    • BaseAgent
      • BaseAgent.default_action_space
      • BaseAgent.load_model()
      • BaseAgent.register_reset()
      • BaseAgent.save_model()
      • BaseAgent.select_action()
      • BaseAgent.update()
  • src.agents.random_agent module
    • RandomAgent
      • RandomAgent.select_action()
  • Module contents
• src.buffers package
  • Submodules
  • src.buffers.PPOBuffer module
    • PPOBuffer
      • PPOBuffer.finish_path()
      • PPOBuffer.instantiate_from_config()
      • PPOBuffer.instantiate_from_config_dict()
      • PPOBuffer.sample_batch()
      • PPOBuffer.schema
      • PPOBuffer.store()
    • discount_cumsum()
  • src.buffers.SimpleReplayBuffer module
    • SimpleReplayBuffer
      • SimpleReplayBuffer.finish_path()
      • SimpleReplayBuffer.instantiate_from_config()
      • SimpleReplayBuffer.instantiate_from_config_dict()
      • SimpleReplayBuffer.sample_batch()
      • SimpleReplayBuffer.schema
      • SimpleReplayBuffer.store()
  • Module contents
• src.config package
  • Submodules
  • src.config.parser module
    • read_config()
  • src.config.schema module
  • src.config.yamlize module
    • ConfigurableDict
      • ConfigurableDict.config
      • ConfigurableDict.name
    • NameToSourcePath
      • NameToSourcePath.agent
      • NameToSourcePath.buffer
      • NameToSourcePath.encoder
      • NameToSourcePath.logger
      • NameToSourcePath.network
      • NameToSourcePath.planner
      • NameToSourcePath.runner
    • create_configurable()
    • create_configurable_from_dict()
    • yamlize()
  • Module contents
• src.encoders package
  • Subpackages
    • src.encoders.transforms package
      • Submodules
      • src.encoders.transforms.preprocessing module
      • Module contents
  • Submodules
  • src.encoders.base module
    • BaseEncoder
      • BaseEncoder.decode()
      • BaseEncoder.encode()
      • BaseEncoder.update()
  • src.encoders.vae module
    • VAE
      • VAE.bottleneck()
      • VAE.decode()
      • VAE.distribution()
      • VAE.encode()
      • VAE.forward()
      • VAE.instantiate_from_config()
      • VAE.instantiate_from_config_dict()
      • VAE.loss()
      • VAE.reparameterize()
      • VAE.representation()
      • VAE.schema
      • VAE.training
      • VAE.update()
  • Module contents
• src.loggers package
  • Submodules
  • src.loggers.FileLogger module
    • FileLogger
      • FileLogger.log()
  • src.loggers.TensorboardLogger module
    • TensorboardLogger
      • TensorboardLogger.log()
  • src.loggers.WanDBLogger module
    • WanDBLogger
      • WanDBLogger.log()
  • src.loggers.base module
    • BaseLogger
      • BaseLogger.log()
  • Module contents
• src.networks package
  • Submodules
  • src.networks.critic module
    • ActivationType
      • ActivationType.ReLU
      • ActivationType.Tanh
    • ActorCritic
      • ActorCritic.act()
      • ActorCritic.instantiate_from_config()
      • ActorCritic.instantiate_from_config_dict()
      • ActorCritic.pi()
      • ActorCritic.schema
      • ActorCritic.training
    • Qfunction
      • Qfunction.forward()
      • Qfunction.instantiate_from_config()
      • Qfunction.instantiate_from_config_dict()
      • Qfunction.schema
      • Qfunction.training
    • SquashedGaussianMLPActor
      • SquashedGaussianMLPActor.forward()
      • SquashedGaussianMLPActor.training
    • Vfunction
      • Vfunction.forward()
      • Vfunction.instantiate_from_config()
      • Vfunction.instantiate_from_config_dict()
      • Vfunction.schema
      • Vfunction.training
    • mlp()
  • src.networks.pets module
    • DynamicsNetwork
      • DynamicsNetwork.forward()
      • DynamicsNetwork.instantiate_from_config()
      • DynamicsNetwork.instantiate_from_config_dict()
      • DynamicsNetwork.predict()
      • DynamicsNetwork.schema
      • DynamicsNetwork.training
    • Ensemble_FC_Layer
      • Ensemble_FC_Layer.forward()
      • Ensemble_FC_Layer.training
  • Module contents
• src.runners package
  • Submodules
  • src.runners.ModelFreeRunner module
    • ModelFreeRunner
      • ModelFreeRunner.checkpoint_model()
      • ModelFreeRunner.eval()
      • ModelFreeRunner.instantiate_from_config()
      • ModelFreeRunner.instantiate_from_config_dict()
      • ModelFreeRunner.run()
      • ModelFreeRunner.save_experiment_state()
      • ModelFreeRunner.schema
  • src.runners.base module
    • BaseRunner
      • BaseRunner.evaluation()
      • BaseRunner.training()
  • Module contents

Submodules

src.constants module

Definition of training constants, and other things which do not need configuration.

Module contents