From 3dd71ae5561c42e9ce6c03273ee5d93be88db06e Mon Sep 17 00:00:00 2001
From: Sanghyun Son <thstkdgus35@snu.ac.kr>
Date: Wed, 21 Aug 2019 11:53:26 +0900
Subject: [PATCH] fix edsr url

---
 src/model/edsr.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/model/edsr.py b/src/model/edsr.py
index 4ed986e..ef4ffb1 100644
--- a/src/model/edsr.py
+++ b/src/model/edsr.py
@@ -23,7 +23,11 @@ class EDSR(nn.Module):
         kernel_size = 3 
         scale = args.scale[0]
         act = nn.ReLU(True)
-        self.url = url['r{}f{}x{}'.format(n_resblocks, n_feats, scale)]
+        url_name = 'r{}f{}x{}'.format(n_resblocks, n_feats, scale)
+        if url_name in url:
+            self.url = url[url_name]
+        else:
+            self.url = None
         self.sub_mean = common.MeanShift(args.rgb_range)
         self.add_mean = common.MeanShift(args.rgb_range, sign=1)
 
-- 
GitLab