Commit ·
4fc35ea
1
Parent(s): cdc8aa3
Update README.md
Browse files
README.md
CHANGED
|
@@ -13,7 +13,12 @@ language:
|
|
| 13 |
## NOTE
|
| 14 |
We have received feedback of people getting poor results on unbalanced datasets. A more robust training script, like scaling
|
| 15 |
the loss and adding weight decay (1e-3 to 1e-5) seems to fix it.
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
If you have any problems using the model, please contact us.
|
|
|
|
| 17 |
Thanks!
|
| 18 |
|
| 19 |
## Introduction
|
|
|
|
| 13 |
## NOTE
|
| 14 |
We have received feedback of people getting poor results on unbalanced datasets. A more robust training script, like scaling
|
| 15 |
the loss and adding weight decay (1e-3 to 1e-5) seems to fix it.
|
| 16 |
+
|
| 17 |
+
Please refer to [this notebook](https://colab.research.google.com/drive/1mYsAk6RgzWsSGmRzcE4mV-UqM9V7_Jes?usp=sharing) to check how performance
|
| 18 |
+
on unbalanced datasets can be improved.
|
| 19 |
+
|
| 20 |
If you have any problems using the model, please contact us.
|
| 21 |
+
|
| 22 |
Thanks!
|
| 23 |
|
| 24 |
## Introduction
|