File size: 616 Bytes
358a820 1919dcb 358a820 |
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 |
{
"model_name": "Magical-1 Star",
"version": "1.0.0",
"sentence_transformer": {
"model_name": "all-MiniLM-L6-v2",
"max_seq_length": 128
},
"classifier": {
"input_size": 384,
"num_classes": 2
},
"training": {
"batch_size": 32,
"num_epochs": 5,
"learning_rate": 2e-5,
"optimizer": "AdamW",
"loss_function": "CrossEntropyLoss"
},
"data": {
"train_test_split": 0.2,
"random_seed": 42
},
"paths": {
"model_save_path": "magical_1_star.safetensors",
"log_dir": "logs"
}
} |