(self, x)
| 796 | mid_channels=intermediate_chn, out_channels=out_ch, depth=rescale_module_depth) |
| 797 | |
| 798 | def forward(self, x): |
| 799 | x = self.encoder(x) |
| 800 | x = self.rescaler(x) |
| 801 | return x |
| 802 | |
| 803 | |
| 804 | class MergedRescaleDecoder(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected