rofunc.learning.RofuncRL.state_encoders.visual_encoders#

1.  Module Contents#

1.1.  Classes#

CNNEncoder

ResnetEncoder

ViTEncoder

1.2.  API#

class rofunc.learning.RofuncRL.state_encoders.visual_encoders.CNNEncoder(cfg)[source]#

Bases: rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder

forward(inputs)[source]#
class rofunc.learning.RofuncRL.state_encoders.visual_encoders.ResnetEncoder(cfg)[source]#

Bases: rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder

forward(inputs)[source]#
class rofunc.learning.RofuncRL.state_encoders.visual_encoders.ViTEncoder(cfg)[source]#

Bases: rofunc.learning.RofuncRL.state_encoders.base_encoders.BaseEncoder

forward(inputs)[source]#