From c425e22beb9cf5420ab6ae2690876db71ef9f355 Mon Sep 17 00:00:00 2001
From: taochenshh <taochen904@gmail.com>
Date: Sun, 28 Jun 2020 18:50:32 -0400
Subject: [PATCH] add pytorch in requirements.txt

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

diff --git a/requirements.txt b/requirements.txt
index c367910..411bf3a 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,4 +8,6 @@ cloudpickle
 gitpython
 tensorboard
 tqdm
+torch==1.4.0
+torchvision==0.5.0
 dataclasses;python_version < '3.7'
\ No newline at end of file