rofunc.learning.RofuncRL.state_encoders.base_encoders#

1.  Module Contents#

1.1.  Classes#

EmptyEncoder

BaseEncoder

MLPEncoder

1.2.  API#

class rofunc.learning.RofuncRL.state_encoders.base_encoders.EmptyEncoder[source]#

Bases: torch.nn.Module

forward(x)[source]#
class rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder(cfg: omegaconf.DictConfig, cfg_name: str = 'state_encoder')[source]#

Bases: torch.nn.Module

set_up()[source]#
freeze_network()[source]#
pre_trained_mode()[source]#
save_ckpt(path: str)[source]#
load_ckpt(path: str)[source]#
class rofunc.learning.RofuncRL.state_encoders.base_encoders.MLPEncoder(cfg, cfg_name)[source]#

Bases: rofunc.learning.RofuncRL.models.base_models.BaseMLP