d3rlpy.models.PixelEncoderFactory¶
- class d3rlpy.models.PixelEncoderFactory(filters=<factory>, feature_size=512, activation='relu', use_batch_norm=False, dropout_rate=None, exclude_last_activation=False, last_activation=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) – Last linear layer size.
activation (str) – Activation function name.
use_batch_norm (bool) – Flag to insert batch normalization layers.
dropout_rate (float) – Dropout probability.
exclude_last_activation (bool) – Flag to exclude activation function at the last layer.
last_activation (str) – Activation function name for the last layer.
Methods
- 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:
PixelEncoderWithAction
- classmethod deserialize(serialized_config)¶
- Parameters:
serialized_config (str) –
- Return type:
TConfig
- classmethod deserialize_from_dict(dict_config)¶
- classmethod from_dict(kvs, *, infer_missing=False)¶
- classmethod from_json(s, *, parse_float=None, parse_int=None, parse_constant=None, infer_missing=False, **kw)¶
- classmethod schema(*, infer_missing=False, only=None, exclude=(), many=False, context=None, load_only=(), dump_only=(), partial=False, unknown=None)¶
- to_dict(encode_json=False)¶
- to_json(*, skipkeys=False, ensure_ascii=True, check_circular=True, allow_nan=True, indent=None, separators=None, default=None, sort_keys=False, **kw)¶
Attributes