mirror of
https://github.com/tcsenpai/agenticSeek.git
synced 2025-06-03 01:30:11 +00:00
33 lines
747 B
JSON
33 lines
747 B
JSON
{
|
|
"config": {
|
|
"batch_size": 32,
|
|
"device_map": "auto",
|
|
"early_stopping_patience": 3,
|
|
"epochs": 10,
|
|
"ewc_lambda": 100.0,
|
|
"gradient_checkpointing": false,
|
|
"learning_rate": 0.0005,
|
|
"max_examples_per_class": 500,
|
|
"max_length": 512,
|
|
"min_confidence": 0.1,
|
|
"min_examples_per_class": 3,
|
|
"neural_weight": 0.2,
|
|
"num_representative_examples": 5,
|
|
"prototype_update_frequency": 50,
|
|
"prototype_weight": 0.8,
|
|
"quantization": null,
|
|
"similarity_threshold": 0.7,
|
|
"warmup_steps": 0
|
|
},
|
|
"embedding_dim": 768,
|
|
"id_to_label": {
|
|
"0": "HIGH",
|
|
"1": "LOW"
|
|
},
|
|
"label_to_id": {
|
|
"HIGH": 0,
|
|
"LOW": 1
|
|
},
|
|
"model_name": "distilbert/distilbert-base-cased",
|
|
"train_steps": 20
|
|
} |