simocorbo's picture
Upload toxicthesis-gemini-3.5-flash-lstm-classification-3 model
b04888a verified
Raw
History Blame Contribute Delete
726 Bytes
{
"experiment_name": "lstm_gemini-3.5-flash_classes3_auto_20260612_145816",
"model_type": "lstm",
"sut": "gemini-3.5-flash",
"num_classes": 3,
"device": "cuda",
"training_config": {
"max_epochs": 200,
"early_stopping_patience": 5,
"batch_size": 200,
"num_workers": 4,
"device": "auto",
"gradient_clip": 1.0,
"checkpoint_dir": "output/checkpoints",
"log_dir": "output/logs",
"gradient_accumulation_steps": 1,
"use_amp": false
},
"model_config": {
"hidden_dim": 512,
"num_layers": 2,
"dropout": 0.3,
"lr": 0.0001,
"bidirectional": true,
"embedding_dim": 300,
"gradient_clip_norm": 1.0,
"num_classes": 3,
"loss_fn": "cross_entropy"
}
}