summaryrefslogtreecommitdiff
path: root/scripts/process.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2018-07-23 16:14:45 +0200
committerJules Laplace <julescarbon@gmail.com>2018-07-23 16:14:45 +0200
commit849acf7ee03f97afb555fc03fee29efc262a8df5 (patch)
tree6a2eddd01fad7e305f80fcf573a373ae407f124d /scripts/process.py
parent0652b8d1cf327c909798cb0fe866edd2c98583f7 (diff)
process
Diffstat (limited to 'scripts/process.py')
-rw-r--r--scripts/process.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/scripts/process.py b/scripts/process.py
index 646c19b..8f5c2f0 100644
--- a/scripts/process.py
+++ b/scripts/process.py
@@ -44,7 +44,7 @@ def load_acoustic_converter():
def load_super_resolution():
sr_config = create_config(sr_model / 'config.json')
- sr_model_paths = base_model.glob('predictor*.npz')
+ sr_model_paths = sr_model.glob('predictor*.npz')
sr_model_path = list(sorted(sr_model_paths, key=extract_number))[-1]
print(sr_model_path)
return SuperResolution(sr_config, sr_model_path, gpu=gpu)