mschonhardt commited on
Commit
13a468b
·
verified ·
1 Parent(s): 19c3992

Upload config.json

Browse files
Files changed (1) hide show
  1. config.json +36 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset_name": "mschonhardt/georges-1913-normalization",
3
+ "test_size": 0.2,
4
+ "val_split": 0.5,
5
+ "random_seed": 42,
6
+ "special_tokens": {
7
+ "<pad>": 0,
8
+ "<sos>": 1,
9
+ "<eos>": 2,
10
+ "<unk>": 3
11
+ },
12
+ "model_parameters": {
13
+ "embedding_dim": 64,
14
+ "hidden_dim": 128,
15
+ "num_layers": 3,
16
+ "dropout": 0.3,
17
+ "max_length": 100
18
+ },
19
+ "training_parameters": {
20
+ "batch_size": 4096,
21
+ "learning_rate": 0.0005,
22
+ "epochs": 10,
23
+ "teacher_forcing_ratio": 0.5,
24
+ "clip_grad_norm": 1.0
25
+ },
26
+ "scheduler_parameters": {
27
+ "scheduler_patience": 2,
28
+ "scheduler_mode": "min",
29
+ "scheduler_verbose": true
30
+ },
31
+ "file_paths": {
32
+ "model_save_path": "normalization_model.pth",
33
+ "vocab_save_path": "vocab.pkl"
34
+ }
35
+ }
36
+