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

Merge branch 'develop'

parents 6f5426b6 3dd71ae5
No related branches found
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