Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
STIX
STIX

Documentation

  • Introduction
  • Installation
  • Tutorials

API reference

  • Overview
    • Core
      • Interpolants
      • Embedders
      • Coupling
      • Criteria
      • Embedded source prior
      • Modality
      • Generative models
      • Sampling generators
      • Utilities
    • Neural networks
      • Network
      • DiT components
      • Context encoders
      • Layers
    • Training
      • Training loop
      • Loss pipeline
      • Time samplers
      • Checkpointing and logging
    • Sampling
      • Adaptive solver
      • Manual solver
      • Guidance
      • Steps
      • Utilities
    • Typing
Back to top
View this page

TrainingΒΆ

The training module contains the training loop, the loss pipeline, time samplers, EMA, and the checkpointing / logging IO handler.

  • Training loop
  • Loss pipeline
  • Time samplers
  • Checkpointing and logging
Next
Training loop
Previous
Layers
Copyright ©
Made with Sphinx and @pradyunsg's Furo