Optimizer

stable_diffusion_trainer::trainer::optimizer

Enum Optimizer

pub enum Optimizer {
Show 19 variants    AdamW,
    AdamW8bit,
    Adafactor,
    DAdaptation,
    DAdaptationGrad,
    DAdaptAdam,
    DAdaptAdan,
    DAdaptAdamIP,
    DAdaptAdamReprint,
    DAdaptLion,
    DAdaptSGD,
    Lion,
    Lion8bit,
    PagedAdamW8bit,
    PagedAdamW32bit,
    PagedLion8bit,
    Prodigy,
    SGDNesterov,
    SGDNesterov8bit,
}

Expand description

The optimizer to use for the training process.

Variants§

AdamW

AdamW optimizer.

AdamW8bit

AdamW 8-bit optimizer.

Adafactor

Adafactor optimizer.

DAdaptation

DAdaptation optimizer.

DAdaptationGrad

DAdaptationGrad optimizer.

DAdaptAdam

DAdaptAdam optimizer.

DAdaptAdan

DAdaptAdan optimizer.

DAdaptAdamIP

DAdaptAdamIP optimizer.

DAdaptAdamReprint

DAdaptAdamReprint optimizer.

DAdaptLion

DAdaptLion optimizer.

DAdaptSGD

DAdaptSGD optimizer.

Lion

Lion optimizer.

Lion8bit

Lion 8-bit optimizer.

PagedAdamW8bit

PagedAdamW 8-bit optimizer.

PagedAdamW32bit

PagedAdamW 32-bit optimizer.

PagedLion8bit

PagedLion 8-bit optimizer.

Prodigy

Prodigy optimizer.

SGDNesterov

SGDNesterov optimizer.

SGDNesterov8bit

SGDNesterov 8-bit optimizer.

Trait Implementations§

impl Debug for Optimizer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

impl<'de> Deserialize<'de> for Optimizer

fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer. Read more

impl Display for Optimizer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

impl Serialize for Optimizer

fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where S: Serializer,

Serialize this value into the given Serde serializer. Read more

Auto Trait Implementations§

impl Freeze for Optimizer

impl RefUnwindSafe for Optimizer

impl Send for Optimizer

impl Sync for Optimizer

impl Unpin for Optimizer

impl UnwindSafe for Optimizer

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> ToString for T
where T: Display + ?Sized,

fn to_string(&self) -> String

Converts the given value to a String. Read more

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<T> DeserializeOwned for T
where T: for<'de> Deserialize<'de>,