Class DiTNoisePredictor<T>.ModelSizes

Namespace

AiDotNet.Diffusion.NoisePredictors

Assembly

AiDotNet.dll

Standard DiT model sizes.

public static class DiTNoisePredictor<T>.ModelSizes

Inheritance

object

DiTNoisePredictor<T>.ModelSizes

Inherited Members

object.Equals(object)

object.Equals(object, object)

object.GetHashCode()

object.GetType()

object.MemberwiseClone()

object.ReferenceEquals(object, object)

object.ToString()

Fields

Base

DiT-B/2: Base model with patch size 2.

public static readonly (int hiddenSize, int numLayers, int numHeads) Base

Field Value

(int TrainSize, int ValidationSize, int TestSize)

Large

DiT-L/2: Large model with patch size 2.

public static readonly (int hiddenSize, int numLayers, int numHeads) Large

Field Value

(int TrainSize, int ValidationSize, int TestSize)

Small

DiT-S/2: Small model with patch size 2.

public static readonly (int hiddenSize, int numLayers, int numHeads) Small

Field Value

(int TrainSize, int ValidationSize, int TestSize)

XLarge

DiT-XL/2: Extra-large model with patch size 2.

public static readonly (int hiddenSize, int numLayers, int numHeads) XLarge

Field Value

(int TrainSize, int ValidationSize, int TestSize)