From 4c9bc8dc7b45082db307b9fb26d88ec3d1b49fdc Mon Sep 17 00:00:00 2001
From: Alessia Marcolini <amarcolini@fbk.eu>
Date: Thu, 16 Apr 2020 11:51:56 +0200
Subject: [PATCH] Add package versions

---
 env.yml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/env.yml b/env.yml
index bb6cde3..26a6305 100644
--- a/env.yml
+++ b/env.yml
@@ -14,8 +14,8 @@ dependencies:
   - pytorch=1.4.0
   - torchvision=0.5
   - tensorboard=2.1
-  - matplotlib
-  - scikit-learn
+  - matplotlib=3.1.3
+  - scikit-learn=0.22.1
   - pyyaml=5.3.1
   - nvidia-apex=0.1