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