d3rlpy.models.encoders.DenseEncoderFactory¶
-
class
d3rlpy.models.encoders.
DenseEncoderFactory
(activation='relu', use_batch_norm=False)[source]¶ DenseNet encoder factory class.
This is an alias for DenseNet architecture proposed in D2RL. This class does exactly same as follows.
from d3rlpy.encoders import VectorEncoderFactory factory = VectorEncoderFactory(hidden_units=[256, 256, 256, 256], use_dense=True)
For now, this only supports vector observations.
References
- 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.VectorEncoder
-
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.VectorEncoderWithAction
Attributes