modelzoo.vision.pytorch.unet.layers.UNetBlock

Classes

UNetBlock

param in_channels The input channel dimension before passing through the block.

previous

modelzoo.vision.pytorch.unet.layers.Encoder.Encoder

next

modelzoo.vision.pytorch.unet.layers.UNetBlock.UNetBlock