From 3438e8e722344bc561159ed21d2f64b1b0b867b0 Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sun, 23 May 2021 17:14:55 +0200
Subject: [PATCH] Add requirements

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

diff --git a/requirements.txt b/requirements.txt
index 8acb9e8..22dee09 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -79,6 +79,9 @@ tensorflow-estimator==2.5.0rc0
 termcolor==1.1.0
 text-unidecode==1.3
 threadpoolctl==2.1.0
+torch==1.8.1
+torchaudio==0.8.1
+torchvision==0.9.1
 tqdm==4.59.0
 typing-extensions==3.7.4.3
 urllib3==1.26.4