Struct DeepSpeedConfig 

pub struct DeepSpeedConfig {
    pub zero_optimization: ZeroOptimizationConfig,
    pub gradient_clipping: Option<f32>,
    pub gradient_accumulation_steps: Option<u32>,
    pub fp16: Option<FP16Config>,
    pub zero_force_ds_cpu_optimizer: Option<bool>,
    pub activation_checkpointing: Option<ActivationCheckpointingConfig>,
}

DeepSpeed configuration compatible with ToRSh

Fields

zero_optimization: ZeroOptimizationConfig

ZeRO optimization configuration

gradient_clipping: Option<f32>

Gradient clipping configuration

gradient_accumulation_steps: Option<u32>

Gradient accumulation steps

fp16: Option<FP16Config>

Mixed precision configuration

zero_force_ds_cpu_optimizer: Option<bool>

CPU offloading configuration

activation_checkpointing: Option<ActivationCheckpointingConfig>

Activation checkpointing

Trait Implementations

impl Clone for DeepSpeedConfig

fn clone(&self) -> DeepSpeedConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for DeepSpeedConfig

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

Formats the value using the given formatter.

impl Default for DeepSpeedConfig

fn default() -> DeepSpeedConfig

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for DeepSpeedConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for DeepSpeedConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.