(self, in_channels, out_channels, kernel_size, bias=True)
| 699 | self.res_scale = res_scale |
| 700 | |
| 701 | def default_conv(self, in_channels, out_channels, kernel_size, bias=True): |
| 702 | return nn.Conv2d(in_channels, out_channels, kernel_size, padding=(kernel_size // 2), bias=bias) |
| 703 | |
| 704 | def forward(self, x): |
| 705 | res = self.body(x) |