tiny_diff.models.unet.layers.UNetSkipLayerMixin

class tiny_diff.models.unet.layers.UNetSkipLayerMixin

Bases: UNetLayerMixin

Mixin for UNet’s layers with skip connections.

__init__()

Methods

__init__()
forward(x, *args, **kwargs)	Forward pass.
residual_forward(h, down_hs, *args, **kwargs)	Forward of the residual blocks.

Attributes

residual_block_cls

Class to use in residual blocks.