ut image.


out_channels (int,  optional, defaults to 3) — Number of channels in the output.


down_block_types (Tuple[str], optional, defaults to  —
obj:("DownEncoderBlock2D",)): Tuple of downsample block types.


up_block_types (Tuple[str], optional, defaults to  —
obj:("UpDecoderBlock2D",)): Tuple of upsample block types.


block_out_channels (Tuple[int], optional, defaults to  —
obj:(64,)): Tuple of block output channels.


act_fn (str, opt