From 85d53eae3694a28da526673ee059c664fc7d971b Mon Sep 17 00:00:00 2001
From: rajeevyasarla <rajeevyasarla19@gmail.com>
Date: Tue, 6 Aug 2019 10:23:15 -0400
Subject: [PATCH] UMSN Files

---
 README.md | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 4e3e0b7..9b7bdaf 100644
--- a/README.md
+++ b/README.md
@@ -10,7 +10,8 @@ We propose a novel multi-stream architecture and training methodology that explo
 ## Prerequisites:
 1. Linux
 2. Python 2 or 3
-3. CPU or NVIDIA GPU + CUDA CuDNN (CUDA 8.0)
+3. Pytorch version 0.4.1
+4. CPU or NVIDIA GPU + CUDA CuDNN (CUDA 8.0)
 
 
 ## To test UMRL: