List of all items

Structs

• MaskedLoss
• StepStats
• TrainError
• Trainer
• checkpoint::LoraCheckpoint
• checkpoint::LoraMetadata
• checkpoint::TrainingCheckpoint
• checkpoint::TrainingState
• data::Batch
• data::DataLoader
• data::TextDataset
• data::TrainingExample
• demo::DemoConfig
• distributed::FileGradSyncConfig
• scaler::GradScaler
• scaler::GradScalerCheckpoint
• scaler::GradScalerStats

Traits

• data::Dataset

Functions

• checkpoint::apply_checkpoint
• checkpoint::load_lora_checkpoint
• checkpoint::load_weights
• checkpoint::save_lora_checkpoint
• data::build_sft_example
• demo::create_demo_data
• demo::print_banner
• demo::print_config
• demo::print_epoch_summary
• demo::print_gpu_info
• demo::print_progress
• demo::run_demo
• distributed::average_grads_file_sync
• distributed::average_host_grads
• distributed::shard_indices
• masked_cross_entropy
• masked_cross_entropy_from_logits