rofunc.learning.RofuncRL.models.base_models

Contents

`rofunc.learning.RofuncRL.models.base_models`#

1. Module Contents#

1.1. Classes#

1.2. API#

class rofunc.learning.RofuncRL.models.base_models.BaseMLP(cfg: omegaconf.DictConfig, input_dim: int, output_dim: int, cfg_name: str)[source]#

Bases: torch.nn.Module

forward(x: torch.Tensor) → torch.Tensor[source]#