autoencoder / config.json
amaye15's picture
Upload AutoEncoder
67d744a
raw
history blame
437 Bytes
{
"architectures": [
"AutoEncoder"
],
"auto_map": {
"AutoConfig": "modeling_autoencoder.AutoEncoderConfig",
"AutoModel": "modeling_autoencoder.AutoEncoder"
},
"bidirectional": false,
"compression_rate": 0.5,
"dropout_rate": 0.1,
"input_dim": 128,
"latent_dim": 64,
"layer_types": "linear",
"model_type": "autoencoder",
"num_layers": 3,
"torch_dtype": "float32",
"transformers_version": "4.35.2"
}