From 9a9bbed0741f33dac6364149c2ee07c1f572b3d6 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Wed, 11 Aug 2021 14:09:24 -0500
Subject: [PATCH] Fixed env.yaml
---
hpvm/env.yaml | 3 +++
1 file changed, 3 insertions(+)
diff --git a/hpvm/env.yaml b/hpvm/env.yaml
index 69adb2635b..7ea2405908 100644
--- a/hpvm/env.yaml
+++ b/hpvm/env.yaml
@@ -4,6 +4,7 @@ channels:
- conda-forge
dependencies:
- Cython
+ - imgaug
- jinja2=2.11
- matplotlib=3.3
- networkx=2.5
@@ -28,3 +29,5 @@ dependencies:
- onnx-simplifier==0.3
- torchnet==0.0.4
- pexpect
+ - mean_average_precision
+ - pytorch-lightning==1.3
\ No newline at end of file
--
GitLab