oa::ml::training::ItTrainingConfig
Configuration for one explicit training loop.
Declaration
pub struct ItTrainingConfig { pub initial_step: u64, pub total_steps: u64, pub steps_per_epoch: u64, pub epoch_steps: Vec<u64>, pub batch_size: u64, pub sequence_length: u64, pub sequence_unit: String, pub source_units_per_sample: f64, pub source_unit: String, pub timer_name: String, pub enable_gpu_timing: bool, }Public Rust declaration.
Public Methods
Associated Functions
Public Data Members
initial_stepu64fieldtotal_stepsu64fieldsteps_per_epochu64fieldepoch_stepsVec<u64>fieldbatch_sizeu64fieldsequence_lengthu64fieldsequence_unitStringfieldsource_units_per_samplef64fieldsource_unitStringfieldtimer_nameStringfieldenable_gpu_timingboolfield