Spaces:
Build error
Build error
Update nets/esrgan.py
Browse files- nets/esrgan.py +1 -1
nets/esrgan.py
CHANGED
@@ -57,7 +57,7 @@ class UpsampleBLock(nn.Module):
|
|
57 |
return x
|
58 |
|
59 |
class Generator(nn.Module):
|
60 |
-
def __init__(self, scale_factor, channels=3, filters=64, num_res_blocks=
|
61 |
super(Generator, self).__init__()
|
62 |
upsample_block_num = int(math.log(scale_factor, 2))
|
63 |
# 第一个卷积层
|
|
|
57 |
return x
|
58 |
|
59 |
class Generator(nn.Module):
|
60 |
+
def __init__(self, scale_factor, channels=3, filters=64, num_res_blocks=6):
|
61 |
super(Generator, self).__init__()
|
62 |
upsample_block_num = int(math.log(scale_factor, 2))
|
63 |
# 第一个卷积层
|