d3rlpy.encoders.DefaultEncoderFactory¶
-
class
d3rlpy.encoders.
DefaultEncoderFactory
(use_batch_norm=False)[source]¶ Default encoder factory class.
This encoder factory returns an encoder based on observation shape.
Parameters: use_batch_norm (bool) – flag to insert batch normalization layers. Methods
-
create
(observation_shape, action_size=None, discrete_action=False)[source]¶ Returns PyTorch’s enocder module.
Parameters: Returns: an enocder object.
Return type: torch.nn.Module
-
get_params
(deep=False)[source]¶ Returns encoder parameters.
Parameters: deep (bool) – flag to deeply copy the parameters. Returns: encoder parameters. Return type: dict
Attributes
-
TYPE
= 'default'¶
-