This abstract class contains all the necessary information about agents and models for setting them up. More...
Public Member Functions | |
def | __init__ (self) |
Static Public Member Functions | |
def | get_prefix_path () |
Gets prefix path for rlpack package, from python installation. More... | |
Data Fields | |
activation_init_args | |
The activation initialization arguments when using in-built models. More... | |
activation_map | |
The mapping between given keyword and PyTorch activation function class. More... | |
agent_args | |
The mapping between given keyword and agent agents' arguments. More... | |
agent_args_default | |
The mapping between given keyword and agent agent's default arguments. More... | |
agent_init_args | |
The mandatory agent initialisation arguments. More... | |
agents | |
The mapping between given keyword and agents models. More... | |
distributions_map | |
The mapping between given keyword and PyTorch Distribution class. More... | |
loss_function_map | |
The mapping between given keyword and PyTorch loss function class. More... | |
lr_scheduler_init_args | |
The LR Scheduler initialization arguments. More... | |
lr_scheduler_map | |
The mapping between given keyword and PyTorch LR Scheduler class. More... | |
mandatory_distributed_agents | |
mandatory_distribution_required_agents | |
mandatory_keys | |
The tuple for mandatory keys (or keyword arguments) always expected. More... | |
model_args | |
The mapping between given keyword and in-built models' arguments. More... | |
model_args_default | |
The mapping between given keyword and in-built model's default arguments. More... | |
model_args_to_optimize | |
The mapping between keyword and agents' model arguments to wrap optimizer with. More... | |
model_init_args | |
The model initialization arguments when using in-built models. More... | |
models | |
The mapping between given keyword and in-built models. More... | |
optimizer_init_args | |
The optimizer initialization arguments for given models. More... | |
optimizer_map | |
The mapping between given keyword and PyTorch optimizer class. More... | |
This abstract class contains all the necessary information about agents and models for setting them up.
def rlpack.utils.base.register.Register.__init__ | ( | self | ) |
Reimplemented in rlpack.utils.setup.Setup, and rlpack.utils.sanity_check.SanityCheck.
|
static |
Gets prefix path for rlpack package, from python installation.
rlpack.utils.base.register.Register.activation_init_args |
The activation initialization arguments when using in-built models.
rlpack.utils.base.register.Register.activation_map |
The mapping between given keyword and PyTorch activation function class.
rlpack.utils.base.register.Register.agent_args |
The mapping between given keyword and agent agents' arguments.
rlpack.utils.base.register.Register.agent_args_default |
The mapping between given keyword and agent agent's default arguments.
rlpack.utils.base.register.Register.agent_init_args |
The mandatory agent initialisation arguments.
rlpack.utils.base.register.Register.agents |
The mapping between given keyword and agents models.
rlpack.utils.base.register.Register.distributions_map |
The mapping between given keyword and PyTorch Distribution class.
rlpack.utils.base.register.Register.loss_function_map |
The mapping between given keyword and PyTorch loss function class.
rlpack.utils.base.register.Register.lr_scheduler_init_args |
The LR Scheduler initialization arguments.
rlpack.utils.base.register.Register.lr_scheduler_map |
The mapping between given keyword and PyTorch LR Scheduler class.
rlpack.utils.base.register.Register.mandatory_distributed_agents |
rlpack.utils.base.register.Register.mandatory_distribution_required_agents |
rlpack.utils.base.register.Register.mandatory_keys |
The tuple for mandatory keys (or keyword arguments) always expected.
rlpack.utils.base.register.Register.model_args |
The mapping between given keyword and in-built models' arguments.
rlpack.utils.base.register.Register.model_args_default |
The mapping between given keyword and in-built model's default arguments.
rlpack.utils.base.register.Register.model_args_to_optimize |
The mapping between keyword and agents' model arguments to wrap optimizer with.
rlpack.utils.base.register.Register.model_init_args |
The model initialization arguments when using in-built models.
rlpack.utils.base.register.Register.models |
The mapping between given keyword and in-built models.
rlpack.utils.base.register.Register.optimizer_init_args |
The optimizer initialization arguments for given models.
rlpack.utils.base.register.Register.optimizer_map |
The mapping between given keyword and PyTorch optimizer class.