From 2df44e85d5f37081000ef4767a0da27ee7357429 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Fri, 19 Mar 2021 17:24:19 -0500
Subject: [PATCH] Adding new Exception for existing directory
---
.../keras/keras_frontend/approxhpvm_translator.py | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py
index 48f469f226..a9e17a4eb8 100644
--- a/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py
+++ b/hpvm/projects/keras/keras_frontend/approxhpvm_translator.py
@@ -1083,6 +1083,10 @@ def getUniquePath(weights_dir):
def createRecursiveDir(target_dir):
+ if os.path.exists(target_dir):
+ print ("Directory = ", target_dir, " exists. Aborting....")
+ sys.exit(1)
+
toks = target_dir.split("/")
for i in range(len(toks)):
path_str = "/".join(toks[0:i+1])
@@ -1103,11 +1107,11 @@ def translate_to_approxhpvm(model,
reload_weights = enable_weights_reload # If set to True, does not dump any weight/input/label files
if not reload_weights:
- weights_dir = getUniquePath(weights_dir)
+ #weights_dir = getUniquePath(weights_dir)
createRecursiveDir(weights_dir)
- src_dir = getUniquePath(src_dir)
+ #src_dir = getUniquePath(src_dir)
createRecursiveDir(src_dir)
dfg = DFG()
--
GitLab