Adding optional working dir to main.py

hpvm/test/epoch_dnn/main.py

@@ -70,6 +70,9 @@ SCP_HOST = "root@espgate.cs.columbia.edu"
 SCP_PWD = "openesp"
 SCP_DST = "~/NV_NVDLA"
 
+if len(sys.argv) > 1:
+    WORKING_DIR = Path(sys.argv[1])
+
 # Reproducibility
 np.random.seed(42)