Spaces:
Runtime error
Runtime error
Upload 2 files
Browse files- model.py +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, (
|
| 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, (
|
| 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:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5053bfccec092aab411a17544cd048bf7f6ec15247b4aa0bf3bc9ed32d37a8b0
|
| 3 |
+
size 69527211
|