Struct TokenBatchingConfig 

pub struct TokenBatchingConfig {
    pub batch_size: usize,
    pub seq_len: usize,
    pub stride: usize,
    pub pad_token: usize,
    pub tail_strategy: TailStrategy,
}

Fields

batch_size: usize

seq_len: usize

stride: usize

pad_token: usize

tail_strategy: TailStrategy

Implementations

impl TokenBatchingConfig

pub fn new(batch_size: usize, seq_len: usize) -> Self

pub fn try_new( batch_size: usize, seq_len: usize, ) -> Result<Self, DataValidationError>

pub fn validate(&self) -> Result<(), DataValidationError>

pub fn with_stride(self, stride: usize) -> Self

pub fn try_with_stride(self, stride: usize) -> Result<Self, DataValidationError>

pub fn with_pad_token(self, pad_token: usize) -> Self

pub fn with_tail_strategy(self, tail_strategy: TailStrategy) -> Self

Trait Implementations

impl Clone for TokenBatchingConfig

fn clone(&self) -> TokenBatchingConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TokenBatchingConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TokenBatchingConfig

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl PartialEq for TokenBatchingConfig

fn eq(&self, other: &TokenBatchingConfig) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for TokenBatchingConfig

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Eq for TokenBatchingConfig

impl StructuralPartialEq for TokenBatchingConfig