stable-diffusion-trainer 0.1.9

Stable Diffusion Trainer
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

//! This module contains the data set types used for training.

pub mod image_data_set;
pub use image_data_set::*;

use crate::prelude::*;

/// A training data set.
#[derive(Debug, Serialize, Deserialize)]
pub struct TrainingDataSet {
    /// The training images.
    pub training: ImageDataSet,
    /// The regularization images.
    pub regularization: Option<ImageDataSet>,
}

impl TrainingDataSet {
    /// Create a new training data set.
    pub fn new(training: ImageDataSet) -> Self {
        let regularization = Default::default();
        TrainingDataSet { training, regularization }
    }

    /// Set the regularization images for the training data set.
    pub fn with_regularization_images(mut self, regularization_images: ImageDataSet) -> Self {
        self.regularization = Some(regularization_images);
        self
    }
}