Update DistilBERT training code

2019-09-05 18:26:14 +00:00
parent f9453d15e5
commit dddd6b9927
3 changed files with 15 additions and 3 deletions
--- a/examples/distillation/requirements.txt
+++ b/examples/distillation/requirements.txt
@@ -1 +1,4 @@
 gitpython==3.0.2
+tensorboard>=1.14.0
+tensorboardX==1.8
+psutil==5.6.3