Preview - rendered architecture diagrams; full validated atlas + dataset coming.
Back to models

UNet2DModel

generative_diffusion-GAN-flow unet2dmodel validation pending

Architecture diagram

Rendered with TorchLens / Graphviz

Open SVG
Unrolled - full
UNet2DModel architecture diagram