underwater45 commited on
Commit
358a820
·
verified ·
1 Parent(s): 699a235

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +27 -0
config.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_name": "Magical-1 Star",
3
+ "version": "1.0.0",
4
+ "sentence_transformer": {
5
+ "model_name": "all-MiniLM-L6-v2",
6
+ "max_seq_length": 128
7
+ },
8
+ "classifier": {
9
+ "input_size": 384,
10
+ "num_classes": 2
11
+ },
12
+ "training": {
13
+ "batch_size": 32,
14
+ "num_epochs": 5,
15
+ "learning_rate": 2e-5,
16
+ "optimizer": "AdamW",
17
+ "loss_function": "CrossEntropyLoss"
18
+ },
19
+ "data": {
20
+ "train_test_split": 0.2,
21
+ "random_seed": 42
22
+ },
23
+ "paths": {
24
+ "model_save_path": "magical_1_star.safetensors",
25
+ "log_dir": "logs"
26
+ }
27
+ }