File size: 894 Bytes
894dcec b3fb2e8 894dcec b3fb2e8 894dcec b3fb2e8 894dcec |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 |
{
"config": {
"batch_size": 32,
"device_map": "auto",
"early_stopping_patience": 3,
"epochs": 10,
"ewc_lambda": 100.0,
"gradient_checkpointing": false,
"learning_rate": 0.001,
"max_examples_per_class": 1000,
"max_length": 512,
"min_confidence": 0.1,
"min_examples_per_class": 3,
"neural_weight": 0.3,
"num_representative_examples": 5,
"prototype_update_frequency": 100,
"prototype_weight": 0.7,
"quantization": null,
"similarity_threshold": 0.6,
"warmup_steps": 0
},
"embedding_dim": 768,
"id_to_label": {
"0": "FOCUSED",
"1": "BALANCED",
"2": "CREATIVE",
"3": "DETERMINISTIC",
"4": "EXPERIMENTAL"
},
"label_to_id": {
"BALANCED": 1,
"CREATIVE": 2,
"DETERMINISTIC": 3,
"EXPERIMENTAL": 4,
"FOCUSED": 0
},
"model_name": "distilbert-base-uncased",
"train_steps": 125
} |