Hezam commited on
Commit
361e612
1 Parent(s): f6980e9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +14 -0
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
  | | |