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

U-ViT-Small (ViT backbone diffusion model with long skip connections between shallow/deep blocks)

history u vit validation pending

Architecture diagram

Rendered with TorchLens / Graphviz

Open SVG
Unrolled - full
U-ViT-Small (ViT backbone diffusion model with long skip connections between shallow/deep blocks) architecture diagram