From bf33be0e278114745c84df90d66e664b16731663 Mon Sep 17 00:00:00 2001
From: Dongsheng Yang <dongshen@ottertune.pdl.cmu.edu>
Date: Sun, 22 Sep 2019 17:23:53 -0400
Subject: [PATCH] add torch in requirements.txt

---
 server/website/requirements.txt | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/server/website/requirements.txt b/server/website/requirements.txt
index 0ff00b6..148b6df 100644
--- a/server/website/requirements.txt
+++ b/server/website/requirements.txt
@@ -19,3 +19,5 @@ scikit-learn==0.19.1
 scipy==1.0.0
 tensorflow==1.6.0
 threadpool==1.3.2
+torch==1.2.0
+torchvision==0.4.0