d3rlpy.encoders.PixelEncoderFactory¶
-
class
d3rlpy.encoders.
PixelEncoderFactory
(filters=None, feature_size=512, activation='relu', use_batch_norm=False)[source]¶ Pixel encoder factory class.
This is the default encoder factory for image observation.
Parameters: 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
= 'pixel'¶
-