ckyrkou commited on
Commit
2dfda0f
·
verified ·
1 Parent(s): 32d941a

Upload 2 files

Browse files
Files changed (2) hide show
  1. model.py +2 -2
  2. model_best.pth +2 -2
model.py CHANGED
@@ -218,10 +218,10 @@ class Net(nn.Module):
218
  self.conv4 = block(128, 256, (5, 5), (1, 1), (2, 2))
219
  self.conv4_1 = block(256, 256, (5, 5), (1, 1), (2, 2))
220
  self.conv5 = block(256, 128, (5, 5), (1, 1), (2, 2))
221
- self.conv6 = block(128, 64, (3, 3), (1, 1), (1, 1))
222
  self.conv7 = block(64, 32, (3, 3), (1, 1), (1, 1))
223
  self.conv7_1 = block(32, 16, (3, 3), (1, 1), (1, 1))
224
- self.conv8 = nn.Conv2d(16, 3, (1, 1), (1, 1), (0, 0))
225
  #self.pixel_shuffle = nn.PixelShuffle(upscale_factor)
226
  #self.relu = nn.ReLU(inplace=True)
227
  #self.upsample = nn.Upsample(scale_factor=2, mode='bicubic')
 
218
  self.conv4 = block(128, 256, (5, 5), (1, 1), (2, 2))
219
  self.conv4_1 = block(256, 256, (5, 5), (1, 1), (2, 2))
220
  self.conv5 = block(256, 128, (5, 5), (1, 1), (2, 2))
221
+ self.conv6 = block(128, 64, (5, 5), (1, 1), (2, 2))
222
  self.conv7 = block(64, 32, (3, 3), (1, 1), (1, 1))
223
  self.conv7_1 = block(32, 16, (3, 3), (1, 1), (1, 1))
224
+ self.conv8 = nn.Conv2d(16, 3, (3, 3), (1, 1), (1, 1))
225
  #self.pixel_shuffle = nn.PixelShuffle(upscale_factor)
226
  #self.relu = nn.ReLU(inplace=True)
227
  #self.upsample = nn.Upsample(scale_factor=2, mode='bicubic')
model_best.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:602198d11d0bbdc5140d258de0d027dbd48f0e82ed567857633a6f4aedbb7164
3
- size 67949739
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5053bfccec092aab411a17544cd048bf7f6ec15247b4aa0bf3bc9ed32d37a8b0
3
+ size 69527211