uprez

2 files changed, 6 insertions, 3 deletions

diff --git a/.gitignore b/.gitignore
index b23047b..6b7d916 100755
--- a/.gitignore
+++ b/.gitignore
@@ -50,4 +50,6 @@ sequences
 checkpoints
 datasets
 thirds
+checkpoints_hdd
+checkpoints_ssd
 
diff --git a/uprez.py b/uprez.py
index 249025d..7255c23 100644
--- a/uprez.py
+++ b/uprez.py
@@ -1,4 +1,5 @@
 import os
+import sys
 import time
 import subprocess
 import argparse
@@ -23,7 +24,7 @@ vsr_duf_results_dir = os.path.abspath('../vsr-duf/results/{}'.format(dataset))
 out_dir = os.path.abspath('./uprez')
 
 subprocess.call([
-  '/home/lens/miniconda3/envs/tf/bin/python',
+  sys.executable,
   '../vsr-duf/crop_frames.py',
   '--in_dir', in_dir,
   '--out_dir', vsr_duf_inputs_dir,
@@ -31,7 +32,7 @@ subprocess.call([
 ])
 
 subprocess.call([
-  '/home/lens/miniconda3/envs/tf/bin/python',
+  sys.executable,
   '../vsr-duf/test.py',
   '--in_dir', vsr_duf_inputs_dir,
   '--out_dir', vsr_duf_results_dir,
@@ -39,7 +40,7 @@ subprocess.call([
 ])
 
 subprocess.call([
-  '/home/lens/miniconda3/envs/tf/bin/python',
+  sys.executable,
   '../vsr-duf/merge_frames.py',
   '--in_dir', vsr_duf_results_dir,
   '--render_dir', out_dir,