lsmille commited on
Commit
2772d1f
·
verified ·
1 Parent(s): e11479d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -1
README.md CHANGED
@@ -20,7 +20,29 @@ It achieves the following results on the evaluation set:
20
 
21
  ## Model description
22
 
23
- More information needed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
 
25
  ## Intended uses & limitations
26
 
 
20
 
21
  ## Model description
22
 
23
+ Trained on single ID token 5K dataset filtered to 10k sequences (30% for test data = 3000)
24
+
25
+ lora_alpha = 64 <--------------
26
+
27
+ lora_dropout = 0.1
28
+
29
+ lora_r = 128
30
+
31
+ epochs = 3
32
+
33
+ learning rate = 3e-4
34
+
35
+ warmup_steps=500
36
+
37
+ gradient_accumulation_steps = 1
38
+
39
+ train_batch = 2
40
+
41
+ eval_batch = 2
42
+
43
+ ALL Linear layers
44
+
45
+ Changed ' token to > <--------------
46
 
47
  ## Intended uses & limitations
48