codelion's picture
Push model using huggingface_hub.
fce041f verified
{
"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
}