Training
Trainer, arguments and pruning:
- TrainingArguments — Training hyperparameters (HF-like)
- Trainer — Generic trainer
- TextPredictionConfig — Config for text prediction
- TextPredictionTrainer — Trainer for text prediction
- PrePruneConfig — Pre-pruning
- PostPruneConfig — Post-pruning
Additional symbols (not in the navigation): load_training_stats, create_model_with_gradiend, GradientTrainingDataset, TextGradientTrainingDataset.