Skip to content
This repository was archived by the owner on Jan 2, 2021. It is now read-only.

Commit 2fd67de

Browse files
authored
Merge pull request #108 from concutere/fix-load-model-absolute-path
fix load_model to use absolute path
2 parents d1ab2b1 + 5248437 commit 2fd67de

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

enhance.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -387,7 +387,7 @@ def load_model(self):
387387
error("Model file with pre-trained convolution layers not found. Download it here...",
388388
"https://github.com/alexjc/neural-enhance/releases/download/v%s/%s"%(__version__, self.get_filename()))
389389
print(' - Loaded file `{}` with trained model.'.format(self.get_filename()))
390-
return pickle.load(bz2.open(self.get_filename(), 'rb'))
390+
return pickle.load(bz2.open(self.get_filename(absolute=True), 'rb'))
391391

392392
def load_generator(self, params):
393393
if len(params) == 0: return

0 commit comments

Comments
 (0)