Enum DistributedStrategy 

pub enum DistributedStrategy {
    DataParallel {
        num_workers: usize,
        batch_size: usize,
    },
    ModelParallel {
        num_shards: usize,
        pipeline_stages: usize,
    },
    Hybrid {
        data_parallel_size: usize,
        model_parallel_size: usize,
    },
}

Distributed training strategy

Variants

DataParallel

Data parallelism - split data across workers

Fields

num_workers: usize

Number of workers

batch_size: usize

Batch size per worker

ModelParallel

Model parallelism - split model across workers

Fields

num_shards: usize

Number of model shards

pipeline_stages: usize

Pipeline stages

Hybrid

Hybrid parallelism - combine data and model parallelism

Fields

data_parallel_size: usize

Data parallel degree

model_parallel_size: usize

Model parallel degree

Trait Implementations

impl Clone for DistributedStrategy

fn clone(&self) -> DistributedStrategy

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DistributedStrategy

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for DistributedStrategy

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for DistributedStrategy

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

where __S: Serializer,

Serialize this value into the given Serde serializer.