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
| 164 | """ |
| 165 | |
| 166 | def __init__( |
| 167 | self, |
| 168 | channels: Optional[int] = None, |
| 169 | out_channels: Optional[int] = None, |
| 170 | use_conv: bool = False, |
| 171 | fir_kernel: Tuple[int, int, int, int] = (1, 3, 3, 1), |
| 172 | ): |
| 173 | super().__init__() |
| 174 | out_channels = out_channels if out_channels else channels |
| 175 | if use_conv: |
| 176 | self.Conv2d_0 = nn.Conv2d(channels, out_channels, kernel_size=3, stride=1, padding=1) |
| 177 | self.fir_kernel = fir_kernel |
| 178 | self.use_conv = use_conv |
| 179 | self.out_channels = out_channels |
| 180 | |
| 181 | def _downsample_2d( |
| 182 | self, |
Callers
nothing calls this directly
Tested by
no test coverage detected