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.