List of all items
Structs
- Accuracy
- AdaBeliefOptimizer
- AdaMaxOptimizer
- AdagradOptimizer
- AdamOptimizer
- AdamPOptimizer
- AdamWOptimizer
- AttentionTransferLoss
- AutocastContext
- AveragingEnsemble
- BCEWithLogitsLoss
- BaggingHelper
- BalancedAccuracy
- BatchCallback
- BatchConfig
- BatchIterator
- BatchReweighter
- BayesianOptimization
- CallbackList
- CheckpointCallback
- ClassBalancedSampler
- CohensKappa
- CompetenceCurriculum
- CompositeAugmenter
- CompositeRegularization
- ConfusionMatrix
- ConsoleLogger
- ConstraintViolationLoss
- ContrastiveLoss
- CosineAnnealingLrScheduler
- CrossEntropyLoss
- CrossValidationResults
- CsvLoader
- CsvLogger
- CurriculumManager
- CurriculumSampler
- CutMixAugmenter
- CutOutAugmenter
- CyclicLrScheduler
- DataPreprocessor
- DataShuffler
- Dataset
- DiceCoefficient
- DiceLoss
- DiscriminativeFineTuning
- DistillationLoss
- DropBlock
- DropPath
- DynamicRangeCalibrator
- EarlyStoppingCallback
- ElasticNetRegularization
- EpisodeSampler
- EpochCallback
- ExpectedCalibrationError
- ExponentialCurriculum
- ExponentialLrScheduler
- ExponentialStochasticDepth
- F1Score
- FeatureDistillationLoss
- FeatureExtractorMode
- FewShotAccuracy
- FileLogger
- FocalLoss
- FocalSampler
- GaussianProcess
- GcConfig
- GcStats
- GlobalPruner
- GradientAccumulationCallback
- GradientAccumulationStats
- GradientCentralization
- GradientCheckpointConfig
- GradientMonitor
- GradientPruner
- GradientScaler
- GradientStats
- GradientSummary
- GridSearch
- GroupLassoRegularization
- HardNegativeMiner
- HingeLoss
- HistogramCallback
- HistogramStats
- HuberLoss
- HyperparamResult
- ImportanceSampler
- IoU
- JsonlLogger
- KFold
- KLDivergenceLoss
- L1Regularization
- L2Regularization
- LabelEncoder
- LabelSmoothingLoss
- LambOptimizer
- LarsOptimizer
- LayerFreezingConfig
- LayerPruningStats
- LearningRateFinder
- LeaveOneOut
- LinearCurriculum
- LinearDropBlockScheduler
- LinearModel
- LinearStochasticDepth
- LionConfig
- LionOptimizer
- LogicalLoss
- LookaheadOptimizer
- LossConfig
- LrRangeTestAnalyzer
- MAML
- MAMLConfig
- MagnitudePruner
- MatchingNetwork
- MatthewsCorrelationCoefficient
- MaxNormRegularization
- MaximumCalibrationError
- MeanAveragePrecision
- MeanIoU
- MemoryBudgetManager
- MemoryEfficientTraining
- MemoryProfilerCallback
- MemorySettings
- MemoryStats
- MetaStats
- MetaTask
- MetricTracker
- MetricsLogger
- MixedPrecisionStats
- MixedPrecisionTrainer
- MixupAugmenter
- MixupLoss
- ModelEMACallback
- ModelSoup
- ModelSummary
- MseLoss
- MultiStepLrScheduler
- MultiTaskLoss
- NAdamOptimizer
- NoAugmentation
- NoamScheduler
- NoiseAugmenter
- NormalizedDiscountedCumulativeGain
- OneCycleLrScheduler
- OneHotEncoder
- OnlineHardExampleMiner
- OptimizerConfig
- OrthogonalRegularization
- PCGrad
- ParameterDifference
- ParameterStats
- PerClassMetrics
- PolyLoss
- PolynomialDecayLrScheduler
- Precision
- ProdigyConfig
- ProdigyOptimizer
- ProfilingCallback
- ProfilingStats
- ProgressiveUnfreezing
- PrototypicalDistance
- PruningConfig
- PruningStats
- QuantizationAwareTraining
- QuantizationConfig
- QuantizationParams
- QuantizedTensor
- Quantizer
- RAdamOptimizer
- RMSpropOptimizer
- RandomErasingAugmenter
- RandomSearch
- Recall
- ReduceLROnPlateauScheduler
- ReduceLrOnPlateauCallback
- Reptile
- ReptileConfig
- RocCurve
- RotationAugmenter
- RuleSatisfactionLoss
- SWACallback
- SamOptimizer
- ScaleAugmenter
- ScheduleFreeAdamW
- ScheduleFreeConfig
- SelfPacedCurriculum
- SgdOptimizer
- SgdrScheduler
- SophiaConfig
- SophiaOptimizer
- SpectralNormalization
- StackingEnsemble
- StepLrScheduler
- StratifiedKFold
- StructuredPruner
- SupportSet
- TaskCurriculum
- TensorBoardLogger
- TimeEstimator
- TimeSeriesSplit
- TopKAccuracy
- Trainer
- TrainerConfig
- TrainingCheckpoint
- TrainingHistory
- TrainingState
- TransferLearningManager
- TripletLoss
- TverskyLoss
- ValidationCallback
- VotingEnsemble
- WarmupCosineLrScheduler
Enums
- AcquisitionFunction
- BitWidth
- CheckpointCompression
- CheckpointStrategy
- CyclicLrMode
- DistanceMetric
- GcStrategy
- GpKernel
- GradClipMode
- GradientScalingStrategy
- Granularity
- HyperparamSpace
- HyperparamValue
- LossScaler
- MiningStrategy
- PlateauMode
- PrecisionMode
- PreprocessingMethod
- QuantizationMode
- ReweightingStrategy
- ShotType
- SophiaVariant
- SoupRecipe
- StructuredPruningAxis
- TaskWeightingStrategy
- TrainError
- VotingMode
Traits
- AutodiffModel
- Callback
- CrossValidationSplit
- CurriculumStrategy
- DataAugmenter
- DynamicModel
- Ensemble
- LoggingBackend
- Loss
- LrScheduler
- MetaLearner
- Metric
- Model
- Optimizer
- Pruner
- Regularizer