Method
__init__
(
self,
channels: Optional[int] = None,
out_channels: Optional[int] = None,
use_conv: bool = False,
fir_kernel: Tuple[int, int, int, int] = (1, 3, 3, 1),
)
Source from the content-addressed store, hash-verified
| 199 | """ |
| 200 | |
| 201 | def __init__( |
| 202 | self, |
| 203 | channels: Optional[int] = None, |
| 204 | out_channels: Optional[int] = None, |
| 205 | use_conv: bool = False, |
| 206 | fir_kernel: Tuple[int, int, int, int] = (1, 3, 3, 1), |
| 207 | ): |
| 208 | super().__init__() |
| 209 | out_channels = out_channels if out_channels else channels |
| 210 | if use_conv: |
| 211 | self.Conv2d_0 = nn.Conv2d(channels, out_channels, kernel_size=3, stride=1, padding=1) |
| 212 | self.use_conv = use_conv |
| 213 | self.fir_kernel = fir_kernel |
| 214 | self.out_channels = out_channels |
| 215 | |
| 216 | def _upsample_2d( |
| 217 | self, |
Callers
nothing calls this directly
Tested by
no test coverage detected