simocorbo's picture
Upload toxicthesis-vicunauc-perspective-classification-2 model
d2e0f30 verified
Raw
History Blame Contribute Delete
659 Bytes
{
"experiment_name": "perspective_vicunaUC_classes2_auto_20260522_141000",
"model_type": "perspective",
"sut": "vicunaUC",
"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": {
"batch_size": 1,
"train_subset_size": 1000,
"val_subset_size": 1000,
"test_subset_size": 1000,
"num_classes": 2,
"loss_fn": "bce"
}
}