diff --git a/requirements.txt b/requirements.txt
index 8dc487e3..1250f800 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,7 +3,8 @@ matplotlib==3.2.2
 numpy==1.18.1
 scipy==1.4.1
 seaborn==0.10.0
-torchvision==0.5.0
+torchvision==0.6.0
 Pillow==7.2.0
 PyYAML==5.3.1
 tqdm==4.46.1
+tensorboard