ut_channels (Tuple[int], optional, defaults to  —
obj:(224, 448, 672, 896)): Tuple of block output channels.


layers_per_block (int, optional, defaults to 2) — The number of layers per block.


mid_block_scale_factor (float, optional, defaults to 1) — The scale factor for the mid block.


downsample_padding (int, optional, defaults to 1) — The padding for the downsample convolution.


act_fn (str, optional, defaults to "silu") — The activation