d3rlpy.models.encoders.PixelEncoderFactory¶
- class d3rlpy.models.encoders.PixelEncoderFactory(filters=None, feature_size=512, activation='relu', use_batch_norm=False, dropout_rate=None)[source]¶
Pixel encoder factory class.
This is the default encoder factory for image observation.
- Parameters
filters (list) – list of tuples consisting with
(filter_size, kernel_size, stride)
. If None,Nature DQN
-based architecture is used.feature_size (int) – the last linear layer size.
activation (str) – activation function name.
use_batch_norm (bool) – flag to insert batch normalization layers.
dropout_rate (float) – dropout probability.
Methods
- create(observation_shape)[source]¶
Returns PyTorch’s state enocder module.
- Parameters
observation_shape (Sequence[int]) – observation shape.
- Returns
an enocder object.
- Return type
d3rlpy.models.torch.encoders.PixelEncoder
- create_with_action(observation_shape, action_size, discrete_action=False)[source]¶
Returns PyTorch’s state-action enocder module.
- Parameters
- Returns
an enocder object.
- Return type
d3rlpy.models.torch.encoders.PixelEncoderWithAction
Attributes