Update README.md

README.md

@@ -13,7 +13,12 @@ language:
 ## NOTE
 We have received feedback of people getting poor results on unbalanced datasets. A more robust training script, like scaling
 the loss and adding weight decay (1e-3 to 1e-5) seems to fix it.
+
+Please refer to [this notebook](https://colab.research.google.com/drive/1mYsAk6RgzWsSGmRzcE4mV-UqM9V7_Jes?usp=sharing) to check how performance
+on unbalanced datasets can be improved.
+
 If you have any problems using the model, please contact us.
+
 Thanks!
 
 ## Introduction