diff --git a/baseline/requirements.txt b/baseline/requirements.txt
index 4944bae..c074b70 100644
--- 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
\ No newline at end of file
+transformers==2.1.1
+tensorboard==2.1.1
+tensorboardx==2.0
\ No newline at end of file