(self, x)
| 140 | self.conv_out = nn.Conv2d(block_out_channels[-1], out_channels, 3, padding=1) |
| 141 | |
| 142 | def __call__(self, x): |
| 143 | x = self.conv_in(x) |
| 144 | |
| 145 | for l in self.down_blocks: |
| 146 | x = l(x) |
| 147 | |
| 148 | x = self.mid_blocks[0](x) |
| 149 | x = self.mid_blocks[1](x) |
| 150 | x = self.mid_blocks[2](x) |
| 151 | |
| 152 | x = self.conv_norm_out(x) |
| 153 | x = nn.silu(x) |
| 154 | x = self.conv_out(x) |
| 155 | |
| 156 | return x |
| 157 | |
| 158 | |
| 159 | class Decoder(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected