revive.algo package

Subpackages

• revive.algo.policy package
  • Submodules
  • revive.algo.policy.base module
    • catch_error()
    • PolicyOperator
      • PolicyOperator.env
      • PolicyOperator.policy
      • PolicyOperator.val_policy
      • PolicyOperator.other_models
      • PolicyOperator.PARAMETER_DESCRIPTION
      • PolicyOperator.get_parameters()
      • PolicyOperator.get_tune_parameters()
      • PolicyOperator.model_creator()
      • PolicyOperator.optimizer_creator()
      • PolicyOperator.data_creator()
      • PolicyOperator.get_ope_dataset()
      • PolicyOperator.venv_test()
      • PolicyOperator.generate_rollout()
      • PolicyOperator.before_train_epoch()
      • PolicyOperator.train_epoch()
      • PolicyOperator.validate()
      • PolicyOperator.train_batch()
      • PolicyOperator.validate_batch()
    • PolicyAlgorithm
      • PolicyAlgorithm.get_train_func()
      • PolicyAlgorithm.get_trainer()
      • PolicyAlgorithm.get_trainable()
      • PolicyAlgorithm.get_parameters()
      • PolicyAlgorithm.get_tune_parameters()
  • revive.algo.policy.ppo module
    • PPOOperator
      • PPOOperator.PARAMETER_DESCRIPTION
      • PPOOperator.model_creator()
      • PPOOperator.optimizer_creator()
      • PPOOperator.data_creator()
      • PPOOperator.train_batch()
      • PPOOperator.bc_train_batch()
      • PPOOperator.ADV()
      • PPOOperator.ppo_step()
  • revive.algo.policy.sac module
    • ReplayBuffer
      • ReplayBuffer.put()
      • ReplayBuffer.__len__()
      • ReplayBuffer.sample()
    • SACOperator
      • SACOperator.PARAMETER_DESCRIPTION
      • SACOperator.policy
      • SACOperator.val_policy
      • SACOperator.model_creator()
      • SACOperator.optimizer_creator()
      • SACOperator.data_creator()
      • SACOperator.setup()
      • SACOperator.train_batch()
      • SACOperator.bc_train_batch()
      • SACOperator.sac()
  • Module contents
• revive.algo.venv package
  • Submodules
  • revive.algo.venv.base module
    • catch_error()
    • VenvOperator
      • VenvOperator.NAME
      • VenvOperator.metric_name
      • VenvOperator.nodes_models_train
      • VenvOperator.other_models_train
      • VenvOperator.nodes_models_val
      • VenvOperator.other_models_val
      • VenvOperator.PARAMETER_DESCRIPTION
      • VenvOperator.get_parameters()
      • VenvOperator.get_tune_parameters()
      • VenvOperator.model_creator()
      • VenvOperator.optimizer_creator()
      • VenvOperator.data_creator()
      • VenvOperator.nan_in_grad()
      • VenvOperator.before_train_epoch()
      • VenvOperator.train_epoch()
      • VenvOperator.validate()
      • VenvOperator.train_batch()
      • VenvOperator.validate_batch()
    • VenvAlgorithm
      • VenvAlgorithm.get_train_func()
      • VenvAlgorithm.get_trainer()
      • VenvAlgorithm.get_trainable()
      • VenvAlgorithm.get_parameters()
      • VenvAlgorithm.get_tune_parameters()
  • revive.algo.venv.bc module
    • BCOperator
      • BCOperator.NAME
      • BCOperator.PARAMETER_DESCRIPTION
      • BCOperator.model_creator()
      • BCOperator.optimizer_creator()
      • BCOperator.data_creator()
      • BCOperator.train_epoch()
      • BCOperator.train_batch()
  • revive.algo.venv.revive module
    • ReviveOperator
      • ReviveOperator.NAME
      • ReviveOperator.matcher_model_creator()
      • ReviveOperator.model_creator()
      • ReviveOperator.data_creator()
      • ReviveOperator.switch_data_loader()
      • ReviveOperator.bc_train_batch()
      • ReviveOperator.train_epoch()
  • revive.algo.venv.revive_p module
    • PPOOperator
      • PPOOperator.NAME
      • PPOOperator.PARAMETER_DESCRIPTION
      • PPOOperator.generator_model_creator()
      • PPOOperator.optimizer_creator()
      • PPOOperator.ADV()
      • PPOOperator.PPO_step()
  • revive.algo.venv.revive_t module
  • revive.algo.venv.template module
    • AlgorithmOperator
      • AlgorithmOperator.NAME
      • AlgorithmOperator.PARAMETER_DESCRIPTION
      • AlgorithmOperator.get_parameters()
      • AlgorithmOperator.get_tune_parameters()
      • AlgorithmOperator.model_creator()
      • AlgorithmOperator.optimizer_creator()
      • AlgorithmOperator.data_creator()
      • AlgorithmOperator.train_epoch()
      • AlgorithmOperator.train_batch()
  • Module contents

Module contents