efaults to None) — optional output processing of UNet.


act_fn (str, optional, defaults to None) — optional activation function in UNet blocks.


norm_num_groups (int, optional, defaults to 8) — group norm member count in UNet blocks.


layers_per_block (int, optional, defaults to 1) — added number of layers in a UNet block.


downsample_each_block (int, optional, defaults to False —
experimental feature for using a UNet without upsampling.


