d3rlpy.models.encoders.DefaultEncoderFactory¶
- class d3rlpy.models.encoders.DefaultEncoderFactory(activation='relu', use_batch_norm=False, dropout_rate=None)[source]¶
Default encoder factory class.
This encoder factory returns an encoder based on observation shape.
- Parameters
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.Encoder
- create_with_action(observation_shape, action_size, discrete_action=False)[source]¶
Returns PyTorch’s state-action enocder module.
Attributes