Returns the encode part of the network by building up a sequence of layers returned by `_get_encode_layer`.
(
self, in_channels: int, channels: Sequence[int], strides: Sequence[int]
)
| 138 | self.decode, _ = self._get_decode_module(self.encoded_channels, decode_channel_list, strides[::-1] or [1]) |
| 139 | |
| 140 | def _get_encode_module( |
| 141 | self, in_channels: int, channels: Sequence[int], strides: Sequence[int] |
| 142 | ) -> tuple[nn.Sequential, int]: |
| 143 | """ |
| 144 | Returns the encode part of the network by building up a sequence of layers returned by `_get_encode_layer`. |
| 145 | """ |
| 146 | encode = nn.Sequential() |
| 147 | layer_channels = in_channels |
| 148 | |
| 149 | for i, (c, s) in enumerate(zip(channels, strides)): |
| 150 | layer = self._get_encode_layer(layer_channels, c, s, False) |
| 151 | encode.add_module(f"encode_{i}", layer) |
| 152 | layer_channels = c |
| 153 | |
| 154 | return encode, layer_channels |
| 155 | |
| 156 | def _get_intermediate_module(self, in_channels: int, num_inter_units: int) -> tuple[nn.Module, int]: |
| 157 | """ |
no test coverage detected