Egrt commited on
Commit
497bc49
1 Parent(s): 623b636

Update nets/esrgan.py

Browse files
Files changed (1) hide show
  1. 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=4):
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
  # 第一个卷积层