Update README.md
Browse files
README.md
CHANGED
@@ -36,6 +36,20 @@ widget:
|
|
36 |
'Art':7,
|
37 |
'Accidents':8
|
38 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
39 |
# # Training parameters
|
40 |
|
41 |
| | |
|
|
|
36 |
'Art':7,
|
37 |
'Accidents':8
|
38 |
}
|
39 |
+
|
40 |
+
|
41 |
+
## Training config
|
42 |
+
|
43 |
+
| | |
|
44 |
+
| :-------------: | :------: |
|
45 |
+
| `batch size` | 128 |
|
46 |
+
| `dropout rate` | 0.1 |
|
47 |
+
| `learning rate` | 0.001 |
|
48 |
+
| `lr schedule` | constant |
|
49 |
+
| `weight decay` | 1e-7 |
|
50 |
+
| `epochs` | 3 |
|
51 |
+
|
52 |
+
|
53 |
# # Training parameters
|
54 |
|
55 |
| | |
|