(self, num_feat, num_grow_ch=32)
| 27 | class RRDB(torch.nn.Module): |
| 28 | |
| 29 | def __init__(self, num_feat, num_grow_ch=32): |
| 30 | super(RRDB, self).__init__() |
| 31 | self.rdb1 = ResidualDenseBlock(num_feat, num_grow_ch) |
| 32 | self.rdb2 = ResidualDenseBlock(num_feat, num_grow_ch) |
| 33 | self.rdb3 = ResidualDenseBlock(num_feat, num_grow_ch) |
| 34 | |
| 35 | def forward(self, x): |
| 36 | out = self.rdb1(x) |
no test coverage detected