From 5248437d068351b4caed3bf8fa1301875fcf0d88 Mon Sep 17 00:00:00 2001 From: concutere Date: Wed, 30 Nov 2016 06:28:18 -0800 Subject: [PATCH] fix load_model to use absolute path --- enhance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enhance.py b/enhance.py index d4ac496..becb851 100755 --- a/enhance.py +++ b/enhance.py @@ -387,7 +387,7 @@ class Model(object): error("Model file with pre-trained convolution layers not found. Download it here...", "https://github.com/alexjc/neural-enhance/releases/download/v%s/%s"%(__version__, self.get_filename())) print(' - Loaded file `{}` with trained model.'.format(self.get_filename())) - return pickle.load(bz2.open(self.get_filename(), 'rb')) + return pickle.load(bz2.open(self.get_filename(absolute=True), 'rb')) def load_generator(self, params): if len(params) == 0: return