autoencoder / config.json
amaye15's picture
Upload AutoEncoder
5836f71
raw
history blame contribute delete
566 Bytes
{
"architectures": [
"AutoEncoder"
],
"auto_map": {
"AutoConfig": "modeling_autoencoder.AutoEncoderConfig",
"AutoModel": "modeling_autoencoder.AutoEncoder"
},
"bidirectional": false,
"bos_token_id": 1,
"compression_rate": 0.5,
"dropout_rate": 0.1,
"embed": false,
"eos_token_id": 2,
"input_dim": 128,
"latent_dim": 64,
"layer_types": "linear",
"max_position": false,
"model_type": "autoencoder",
"num_layers": 3,
"pad_token_id": 0,
"torch_dtype": "float32",
"transformers_version": "4.35.2",
"vocab_size": false
}