Update README.md
Browse files
README.md
CHANGED
@@ -28,9 +28,15 @@ modules_to_save: ["classifier"]
|
|
28 |
|
29 |
### Training config
|
30 |
class: AdamW
|
|
|
31 |
betas: (0.9, 0.98)
|
|
|
32 |
weight_decay: 0.01
|
|
|
33 |
learning rate: 1e-4
|
|
|
34 |
epoch: 50
|
|
|
35 |
batch size: 2
|
|
|
36 |
precision: 16-mixed
|
|
|
28 |
|
29 |
### Training config
|
30 |
class: AdamW
|
31 |
+
|
32 |
betas: (0.9, 0.98)
|
33 |
+
|
34 |
weight_decay: 0.01
|
35 |
+
|
36 |
learning rate: 1e-4
|
37 |
+
|
38 |
epoch: 50
|
39 |
+
|
40 |
batch size: 2
|
41 |
+
|
42 |
precision: 16-mixed
|