Module stable_diffusion_trainer::trainer

The Trainer module contains the training configuration and the training process.

Re-exports

• pub use training::*;
• pub use output::*;
• pub use optimizer::*;
• pub use scheduler::*;
• pub use parameters::*;

Modules

• optimizer
  Optimizer module for the trainer.
• output
  Trainer’s output.
• parameters
  Trainer’s parameters.
• scheduler
  Learning rate scheduler module.
• training
  The training configuration for the training process.

Structs

• Trainer
  The Trainer structure.