malib.rl package

Subpackages

• malib.rl.a2c package
  • Submodules
  • malib.rl.a2c.config module
  • malib.rl.a2c.policy module
    • A2CPolicy
      • A2CPolicy.value_function()
  • malib.rl.a2c.trainer module
    • A2CTrainer
      • A2CTrainer.post_process()
      • A2CTrainer.setup()
      • A2CTrainer.train()
• malib.rl.coma package
  • Submodules
  • malib.rl.coma.critic module
    • COMADiscreteCritic
      • COMADiscreteCritic.forward()
      • COMADiscreteCritic.training
  • malib.rl.coma.trainer module
    • COMATrainer
      • COMATrainer.create_joint_action()
      • COMATrainer.post_process()
      • COMATrainer.setup()
      • COMATrainer.train()
      • COMATrainer.train_critic()
• malib.rl.common package
  • Submodules
  • malib.rl.common.misc module
    • gumbel_softmax()
    • masked_logits()
    • onehot_from_logits()
    • sample_gumbel()
    • soft_update()
    • softmax()
  • malib.rl.common.policy module
    • Policy
      • Policy.actor
      • Policy.compute_action()
      • Policy.coordinate()
      • Policy.copy()
      • Policy.critic
      • Policy.custom_config
      • Policy.deregister_state()
      • Policy.device
      • Policy.get_initial_state()
      • Policy.load()
      • Policy.load_state_dict()
      • Policy.model_config
      • Policy.parameters()
      • Policy.preprocessor
      • Policy.register_state()
      • Policy.registered_networks
      • Policy.reset()
      • Policy.save()
      • Policy.state_dict()
      • Policy.target_actor
      • Policy.target_critic
      • Policy.to()
      • Policy.update_parameters()
    • SimpleObject
      • SimpleObject.load_state_dict()
      • SimpleObject.state_dict()
  • malib.rl.common.trainer module
    • Trainer
      • Trainer.counter
      • Trainer.parameters()
      • Trainer.policy
      • Trainer.post_process()
      • Trainer.reset()
      • Trainer.setup()
      • Trainer.step_counter()
      • Trainer.train()
      • Trainer.training_config
• malib.rl.discrete_sac package
  • Submodules
  • malib.rl.discrete_sac.policy module
  • malib.rl.discrete_sac.trainer module
• malib.rl.dqn package
  • Submodules
  • malib.rl.dqn.config module
  • malib.rl.dqn.policy module
    • DQNPolicy
      • DQNPolicy.compute_action()
      • DQNPolicy.eps
      • DQNPolicy.load()
      • DQNPolicy.parameters()
      • DQNPolicy.reset()
      • DQNPolicy.save()
      • DQNPolicy.value_function()
  • malib.rl.dqn.trainer module
    • DQNTrainer
      • DQNTrainer.post_process()
      • DQNTrainer.setup()
      • DQNTrainer.train()
• malib.rl.maddpg package
  • Submodules
  • malib.rl.maddpg.loss module
  • malib.rl.maddpg.trainer module
• malib.rl.mappo package
  • Submodules
  • malib.rl.mappo.config module
  • malib.rl.mappo.policy module
  • malib.rl.mappo.trainer module
• malib.rl.pg package
  • Submodules
  • malib.rl.pg.config module
  • malib.rl.pg.policy module
    • PGPolicy
      • PGPolicy.compute_action()
      • PGPolicy.value_function()
  • malib.rl.pg.trainer module
    • PGTrainer
      • PGTrainer.post_process()
      • PGTrainer.setup()
      • PGTrainer.train()
• malib.rl.ppo package
  • Submodules
  • malib.rl.ppo.policy module
  • malib.rl.ppo.trainer module
• malib.rl.qmix package
  • Submodules
  • malib.rl.qmix.q_mixer module
  • malib.rl.qmix.trainer module
• malib.rl.random package
  • Submodules
  • malib.rl.random.config module
  • malib.rl.random.policy module
    • RandomPolicy
  • malib.rl.random.random_trainer module
    • RandomTrainer
• malib.rl.sac package
  • Submodules
  • malib.rl.sac.policy module
  • malib.rl.sac.trainer module