thebogko commited on
Commit
d3f7498
1 Parent(s): cb20471
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -113,7 +113,7 @@ Gridspace search was applied to find the best learning rate, epoch number, weigh
113
  ```
114
  gridSpace = {
115
  'batch_size': [4, 8],
116
- 'lr_rate': [0.0002, 2e-4, 2e-5],
117
  'w_decay': [0.1, 0.01, 0.001]
118
  }
119
  ```
@@ -121,9 +121,9 @@ gridSpace = {
121
  Along with an epoch number from 1 to 16.
122
 
123
  The chosen setup at the end of experimentation stage was chosen to be:
124
- 1) **batch_size**: 4
125
  2) **learning_rate**: 0.0002
126
- 3) **wight_decay**: 0.001
127
  4) **epoch number**: 4
128
 
129
  This gridspace search was performed 3 separate times, and it resulted in the lowest average validation loss of 0.01431.
 
113
  ```
114
  gridSpace = {
115
  'batch_size': [4, 8],
116
+ 'lr_rate': [0.002, 0.0002, 0.00002],
117
  'w_decay': [0.1, 0.01, 0.001]
118
  }
119
  ```
 
121
  Along with an epoch number from 1 to 16.
122
 
123
  The chosen setup at the end of experimentation stage was chosen to be:
124
+ 1) **batch_size**: 8
125
  2) **learning_rate**: 0.0002
126
+ 3) **wight_decay**: 0.01
127
  4) **epoch number**: 4
128
 
129
  This gridspace search was performed 3 separate times, and it resulted in the lowest average validation loss of 0.01431.