Added tensorboard python modules to requirements.txt

3 years ago · 3fc15d3dd2
parent 2d94f036da
commit 3fc15d3dd2
1 changed files with 3 additions and 1 deletions
--- a/baseline/requirements.txt
+++ b/baseline/requirements.txt
@ -1,4 +1,6 @@
 numpy==1.17.4
 torch==1.6.0
 tqdm==4.38.0
-transformers==2.1.1
+transformers==2.1.1
+tensorboard==2.1.1
+tensorboardx==2.0