core#
Source code: tianshou/highlevel/module/core.py
- init_linear_orthogonal(module: Module) None [source]#
Applies orthogonal initialization to linear layers of the given module and sets bias weights to 0.
- Parameters:
module – the module whose submodules are to be processed
- class ModuleFactory[source]#
Bases:
ABC
Represents a factory for the creation of a torch module given an environment and target device.
- abstract create_module(envs: Environments, device: str | device) Module [source]#