simocorbo's picture
Upload toxicthesis-vicunauc-perspective-classification-4 model
8bcee9e verified
Raw
History Blame Contribute Delete
669 Bytes
{
"experiment_name": "perspective_vicunaUC_classes4_auto_20260522_150009",
"model_type": "perspective",
"sut": "vicunaUC",
"num_classes": 4,
"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": {
"batch_size": 1,
"train_subset_size": 1000,
"val_subset_size": 1000,
"test_subset_size": 1000,
"num_classes": 4,
"loss_fn": "cross_entropy"
}
}