simocorbo's picture
Upload toxicthesis-gemini-3.5-flash-linear-classification-2 model
3a4a5ce verified
Raw
History Blame Contribute Delete
738 Bytes
{
"experiment_name": "linear_gemini-3.5-flash_classes2_auto_20260612_100038",
"model_type": "linear",
"sut": "gemini-3.5-flash",
"num_classes": 2,
"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": {
"input_dim": 300,
"l2_lambda": 0.01,
"learning_rate": 0.01,
"max_epochs": 100,
"batch_size": 512,
"dropout": 0.0,
"use_amp": false,
"use_scaler": true,
"num_classes": 2,
"loss_fn": "bce"
}
}