From 1d02231bf7b8d9e9a4f922e254e3d2973ccb1c51 Mon Sep 17 00:00:00 2001
From: Filip Izydorczyk <filip.izydorczyk@protonmail.com>
Date: Fri, 4 Jun 2021 16:12:01 +0200
Subject: [PATCH] torch requirements update

---
 requirements.txt | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 4a84621..1c60281 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,5 @@ matplotlib
 seaborn
 sacred
 pymongo
-mlflow
\ No newline at end of file
+mlflow
+torchvision
\ No newline at end of file