update params

README.md

@@ -75,11 +75,11 @@ Find the v1 (TensorFlow) model on [this page](https://github.com/FredZhang7/tfjs
 | Framework  | PyTorch   |  TensorFlow   |
 | Dataset Size  |  3.0M |  2.68M   |
 | Train Split | 80% English<br>20% English + 100% Multilingual |  None  |
-| English Train Accuracy  |  99.4% |  N/A (≈97.5%)  |
+| English Train Accuracy  |  99.5% |  N/A (≈97.5%)  |
 | Other Train Accuracy  | 96.7%  |  96.6%  |
 | Final Val Accuracy  |  95.2%  |  94.6%  |
 | Languages |  55  |  N/A (≈35)  |
-| Hyperparameters  | maxlen=208<br>padding='max_length'<br>batch_size=112<br>optimizer=Adam<br>learning_rate=1e-5<br>loss=BCEWithLogitsLoss()  |  maxlen=192<br>padding='max_length'<br>batch_size=16<br>optimizer=Adam<br>learning_rate=1e-5<br>loss="binary_crossentropy"  |
+| Hyperparameters  | maxlen=208<br>padding='max_length'<br>batch_size=112<br>optimizer=AdamW<br>learning_rate=1e-5<br>loss=BCEWithLogitsLoss()  |  maxlen=192<br>padding='max_length'<br>batch_size=16<br>optimizer=Adam<br>learning_rate=1e-5<br>loss="binary_crossentropy"  |
 | Training Stopped |  7/18/2023  |  9/05/2022  |
 
 <br>