protomotions.agents.common.fsq_config module#

Configuration for finite-scalar-quantized autoencoders.

class protomotions.agents.common.fsq_config.FSQAutoEncoderConfig( _target_='protomotions.agents.common.fsq.FSQAutoEncoder', in_keys=<factory>, out_keys=<factory>, encoder_out_keys=<factory>, decoder_out_keys=<factory>, encoder=<factory>, decoder=<factory>, latent_key='latent', num_fsq_levels=7, num_fsq_scalars=24, )[source]#

Bases: AutoEncoderConfig

Finite scalar quantization autoencoder configuration.

Attributes:: in_keys: Input keys. out_keys: Output keys. encoder_out_keys: Encoder output keys. decoder_out_keys: Decoder output keys. encoder: Encoder network configuration. decoder: Decoder network configuration. latent_key: TensorDict key used to pass latents to decoder. num_fsq_levels: Number of quantization levels. Must be odd. num_fsq_scalars: Number of FSQ scalar code dimensions.

num_fsq_levels: int = 7#

num_fsq_scalars: int = 24#

__init__( _target_='protomotions.agents.common.fsq.FSQAutoEncoder', in_keys=<factory>, out_keys=<factory>, encoder_out_keys=<factory>, decoder_out_keys=<factory>, encoder=<factory>, decoder=<factory>, latent_key='latent', num_fsq_levels=7, num_fsq_scalars=24, )#