def __init__(self, config, img_size):
    super().__init__()
    self.config = config
    self.down_factor = config.down_factor
    head_channels = config.conv_first_channel
    self.img_size = img_size
    self.conv_more = Conv3dReLU(config.hidden_size, head_channels, kernel_size=3, padding=1, use_batchnorm=True)
    decoder_channels = config.decoder_channels
    in_channels = ([head_channels] + list(decoder_channels[:(- 1)]))
    out_channels = decoder_channels
    self.patch_size = _triple(config.patches['size'])
    skip_channels = self.config.skip_channels
    blocks = [DecoderBlock(in_ch, out_ch, sk_ch) for (in_ch, out_ch, sk_ch) in zip(in_channels, out_channels, skip_channels)]
    self.blocks = nn.ModuleList(blocks)
def __init__(self, config, img_size):
    super().__init__()
    self.config = config
    self.down_factor = config.down_factor
    head_channels = config.conv_first_channel
    self.img_size = img_size
    self.conv_more = Conv3dReLU(config.hidden_size, head_channels, kernel_size=3, padding=1, use_batchnorm=True)
    decoder_channels = config.decoder_channels
    in_channels = ([head_channels] + list(decoder_channels[:(- 1)]))
    out_channels = decoder_channels
    self.patch_size = _triple(config.patches['size'])
    skip_channels = self.config.skip_channels
    blocks = [DecoderBlock(in_ch, out_ch, sk_ch) for (in_ch, out_ch, sk_ch) in zip(in_channels, out_channels, skip_channels)]
    self.blocks = nn.ModuleList(blocks)