Skip to content
Snippets Groups Projects
Commit 9d3bb0ec authored by Sanghyun Son's avatar Sanghyun Son
Browse files

Merge branch 'develop'

parents 6f5426b6 3dd71ae5
Branches
No related tags found
1 merge request!1Jan 09, 2018 updates
...@@ -23,7 +23,11 @@ class EDSR(nn.Module): ...@@ -23,7 +23,11 @@ class EDSR(nn.Module):
kernel_size = 3 kernel_size = 3
scale = args.scale[0] scale = args.scale[0]
act = nn.ReLU(True) 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.sub_mean = common.MeanShift(args.rgb_range)
self.add_mean = common.MeanShift(args.rgb_range, sign=1) self.add_mean = common.MeanShift(args.rgb_range, sign=1)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment