(self, out_channels: int)
| 147 | return up_layers, up_samples |
| 148 | |
| 149 | def _make_final_conv(self, out_channels: int): |
| 150 | return nn.Sequential( |
| 151 | get_norm_layer(name=self.norm, spatial_dims=self.spatial_dims, channels=self.init_filters), |
| 152 | self.act_mod, |
| 153 | get_conv_layer(self.spatial_dims, self.init_filters, out_channels, kernel_size=1, bias=True), |
| 154 | ) |
| 155 | |
| 156 | def encode(self, x: torch.Tensor) -> tuple[torch.Tensor, list[torch.Tensor]]: |
| 157 | x = self.convInit(x) |
no test coverage detected