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"
    }
}