tiny_diff.modules.residual.blocks

Classes

ERABlock(in_channels, out_channels, ...)

Residual block with self attention at the end.

ERARBlock(in_channels, out_channels, ...)

Residual layer with attention in the middle.

ERBlock(embed_channels, out_channels[, ...])

Block for residual layer with embeddings.

RABlock(in_channels, out_channels, ...)

Residual block with self attention at the end.

RABlockABC(in_channels, out_channels, ...)

Residual block with self attention at the end.

RBlock(in_channels, out_channels[, drop_p, ...])

Block for residual layers.