amaye15's picture
Training in progress, epoch 1
fabf416 verified
{
"activation": "gelu",
"architectures": [
"AutoencoderForReconstruction"
],
"auto_map": {
"AutoConfig": "configuration_autoencoder.AutoencoderConfig",
"AutoModel": "modeling_autoencoder.AutoencoderForReconstruction"
},
"autoencoder_type": "classic",
"beta": 1.0,
"bidirectional": true,
"dropout_rate": 0.1,
"flow_coupling_layers": 2,
"hidden_dims": [
64,
32
],
"input_dim": 20,
"latent_dim": 16,
"learn_inverse_preprocessing": true,
"model_type": "autoencoder",
"noise_factor": 0.1,
"num_layers": 2,
"preprocessing_hidden_dim": 32,
"preprocessing_num_layers": 2,
"preprocessing_type": "robust_scaler",
"reconstruction_loss": "mse",
"rnn_type": "lstm",
"sequence_length": null,
"teacher_forcing_ratio": 0.5,
"tie_weights": false,
"torch_dtype": "float32",
"transformers_version": "4.55.2",
"use_batch_norm": true,
"use_learnable_preprocessing": true
}