(self, x)
| 715 | mid_channels=intermediate_chn, out_channels=out_ch, depth=rescale_module_depth) |
| 716 | |
| 717 | def forward(self, x): |
| 718 | x = self.encoder(x) |
| 719 | x = self.rescaler(x) |
| 720 | return x |
| 721 | |
| 722 | |
| 723 | class MergedRescaleDecoder(nn.Module): |
nothing calls this directly
no outgoing calls
no test coverage detected