lsmille commited on
Commit
272bf56
·
verified ·
1 Parent(s): 15cb3a1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +22 -2
README.md CHANGED
@@ -20,7 +20,27 @@ It achieves the following results on the evaluation set:
20
 
21
  ## Model description
22
 
23
- More information needed
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
24
 
25
  ## Intended uses & limitations
26
 
@@ -28,7 +48,7 @@ More information needed
28
 
29
  ## Training and evaluation data
30
 
31
- More information needed
32
 
33
  ## Training procedure
34
 
 
20
 
21
  ## Model description
22
 
23
+ lora_alpha = 32
24
+
25
+ lora_dropout = 0.05
26
+
27
+ lora_r = 16
28
+
29
+ epochs = 3
30
+
31
+ learning rate = 3e-4
32
+
33
+ warmup_steps=0.5
34
+
35
+ gradient_accumulation_steps = 8
36
+
37
+ train_batch = 1
38
+
39
+ eval_batch = 1
40
+
41
+ Training of only last 40 linear modules [120:160] instead of [0:160] <------
42
+
43
+ This changes the # of trainable params to 8,914,944
44
 
45
  ## Intended uses & limitations
46
 
 
48
 
49
  ## Training and evaluation data
50
 
51
+ look at files
52
 
53
  ## Training procedure
54