{ | |
"config": { | |
"batch_size": 4, | |
"device_map": "auto", | |
"early_stopping_patience": 2, | |
"epochs": 5, | |
"ewc_lambda": 100.0, | |
"gradient_checkpointing": false, | |
"learning_rate": 2e-05, | |
"max_examples_per_class": 100, | |
"max_length": 2048, | |
"min_confidence": 0.1, | |
"min_examples_per_class": 3, | |
"neural_weight": 0.3, | |
"num_representative_examples": 5, | |
"prototype_update_frequency": 10, | |
"prototype_weight": 0.7, | |
"quantization": null, | |
"similarity_threshold": 0.6, | |
"warmup_steps": 0 | |
}, | |
"embedding_dim": 1024, | |
"id_to_label": { | |
"0": "NOT_HALLUCINATED", | |
"1": "HALLUCINATED" | |
}, | |
"label_to_id": { | |
"HALLUCINATED": 1, | |
"NOT_HALLUCINATED": 0 | |
}, | |
"model_name": "answerdotai/ModernBERT-large", | |
"train_steps": 36 | |
} |