File size: 1,028 Bytes
f5762fa | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | {
"model_type": "vae",
"architectures": ["VariationalAutoencoder"],
"model_name": "VAE-Banking77-OpenIntentClassification",
"input_dim": 768,
"hidden_dim": 256,
"latent_dim": 64,
"encoder_config": {
"layers": [
{"type": "Linear", "in_features": 768, "out_features": 256},
{"type": "ReLU"}
],
"mu_layer": {"type": "Linear", "in_features": 256, "out_features": 64},
"logvar_layer": {"type": "Linear", "in_features": 256, "out_features": 64}
},
"decoder_config": {
"layers": [
{"type": "Linear", "in_features": 64, "out_features": 256},
{"type": "ReLU"},
{"type": "Linear", "in_features": 256, "out_features": 768}
]
},
"training_config": {
"dataset": "PolyAI/banking77",
"loss_weights": {
"reconstruction_loss": 0.5,
"kl_divergence": 0.5
},
"validation_metric": "reconstruction_loss"
},
"library_name": "pytorch",
"torch_dtype": "float32",
"transformers_version": null,
"created_date": "2025-08-11"
}
|