Torchium
User Guide
Quickstart Guide
Advanced Usage Guide
Domain-Specific Usage Guide
Performance Guide
API Reference
Optimizers API
Loss Functions API
Examples
Computer Vision Examples
Natural Language Processing Examples
Generative Models Examples
Optimization Comparison Examples
Benchmarks Examples
Torchium
Index
Index
_
|
A
|
B
|
C
|
D
|
E
|
F
|
G
|
H
|
I
|
K
|
L
|
M
|
N
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
Y
|
Z
_
__init__() (torchium.losses.AdaptiveLogSoftmaxWithLoss method)
(torchium.losses.AlphaIoULoss method)
(torchium.losses.AngularMetricLoss method)
(torchium.losses.ArcFaceMetricLoss method)
(torchium.losses.BCELoss method)
(torchium.losses.BCEWithLogitsLoss method)
(torchium.losses.BetaTCVAELoss method)
(torchium.losses.BLEULoss method)
(torchium.losses.BoundaryLoss method)
(torchium.losses.CAGradLoss method)
(torchium.losses.CIoULoss method)
(torchium.losses.ClassBalancedLoss method)
(torchium.losses.ContentLoss method)
(torchium.losses.ContrastiveLoss method)
(torchium.losses.ContrastiveMetricLoss method)
(torchium.losses.CosFaceMetricLoss method)
(torchium.losses.CosineEmbeddingLoss method)
(torchium.losses.CRFLoss method)
(torchium.losses.CrossEntropyLoss method)
(torchium.losses.CTCLoss method)
(torchium.losses.DiceLoss method)
,
[1]
(torchium.losses.DIoULoss method)
(torchium.losses.DTWLoss method)
(torchium.losses.DynamicLossBalancing method)
(torchium.losses.EIoULoss method)
(torchium.losses.ELBOLoss method)
(torchium.losses.FocalDetectionLoss method)
(torchium.losses.FocalLoss method)
(torchium.losses.FocalTverskyLoss method)
(torchium.losses.GANLoss method)
(torchium.losses.GaussianNLLLoss method)
(torchium.losses.GIoULoss method)
(torchium.losses.GradNormLoss method)
(torchium.losses.HingeEmbeddingLoss method)
(torchium.losses.IoULoss method)
(torchium.losses.KLDivLoss method)
(torchium.losses.L1Loss method)
(torchium.losses.LabelSmoothingLoss method)
(torchium.losses.LogCoshLoss method)
(torchium.losses.LovaszLoss method)
(torchium.losses.LPIPSLoss method)
(torchium.losses.MarginRankingLoss method)
(torchium.losses.MSSSIMLoss method)
(torchium.losses.MultiLabelMarginLoss method)
(torchium.losses.MultiLabelSoftMarginLoss method)
(torchium.losses.MultiMarginLoss method)
(torchium.losses.MultiTaskLoss method)
(torchium.losses.NLLLoss method)
(torchium.losses.NPairLoss method)
(torchium.losses.PCGradLoss method)
(torchium.losses.PerceptualLoss method)
(torchium.losses.PerplexityLoss method)
(torchium.losses.PoissonNLLLoss method)
(torchium.losses.ProxyAnchorLoss method)
(torchium.losses.ProxyNCALoss method)
(torchium.losses.QuadrupletLoss method)
(torchium.losses.QuantileLoss method)
(torchium.losses.SoftMarginLoss method)
(torchium.losses.SpectralLoss method)
(torchium.losses.SphereFaceLoss method)
(torchium.losses.SSIMLoss method)
(torchium.losses.StructuredPredictionLoss method)
(torchium.losses.StyleLoss method)
(torchium.losses.TotalVariationLoss method)
(torchium.losses.TripletMarginLoss method)
(torchium.losses.TripletMarginWithDistanceLoss method)
(torchium.losses.TripletMetricLoss method)
(torchium.losses.TverskyLoss method)
,
[1]
(torchium.losses.UncertaintyWeightingLoss method)
(torchium.losses.VGGLoss method)
(torchium.losses.Word2VecLoss method)
(torchium.optimizers.AdaBelief method)
(torchium.optimizers.AdaBound method)
(torchium.optimizers.Adadelta method)
(torchium.optimizers.AdaFactor method)
(torchium.optimizers.Adagrad method)
(torchium.optimizers.AdaHessian method)
,
[1]
(torchium.optimizers.Adam method)
(torchium.optimizers.AdamP method)
(torchium.optimizers.AdamW method)
(torchium.optimizers.Apollo method)
(torchium.optimizers.ASAM method)
(torchium.optimizers.CMAES method)
(torchium.optimizers.DifferentialEvolution method)
(torchium.optimizers.FTRL method)
(torchium.optimizers.GeneticAlgorithm method)
(torchium.optimizers.GradientCentralization method)
(torchium.optimizers.GradNorm method)
(torchium.optimizers.GSAM method)
(torchium.optimizers.HeavyBall method)
(torchium.optimizers.KFAC method)
(torchium.optimizers.LAMB method)
(torchium.optimizers.LARS method)
(torchium.optimizers.LBFGS method)
(torchium.optimizers.Lion method)
(torchium.optimizers.LookSAM method)
(torchium.optimizers.MADGRAD method)
(torchium.optimizers.NAdam method)
(torchium.optimizers.NaturalGradient method)
(torchium.optimizers.NovoGrad method)
(torchium.optimizers.ParticleSwarmOptimization method)
(torchium.optimizers.PCGrad method)
(torchium.optimizers.QuantumAnnealing method)
(torchium.optimizers.RAdam method)
(torchium.optimizers.Ranger method)
(torchium.optimizers.RMSprop method)
(torchium.optimizers.Rprop method)
(torchium.optimizers.SAM method)
(torchium.optimizers.SGD method)
(torchium.optimizers.Shampoo method)
(torchium.optimizers.SM3 method)
(torchium.optimizers.SparseAdam method)
(torchium.optimizers.WSAM method)
(torchium.optimizers.Yogi method)
A
AdaBelief (class in torchium.optimizers)
AdaBound (class in torchium.optimizers)
Adadelta (class in torchium.optimizers)
AdaFactor (class in torchium.optimizers)
Adagrad (class in torchium.optimizers)
AdaHessian (class in torchium.optimizers)
,
[1]
Adam (class in torchium.optimizers)
AdamP (class in torchium.optimizers)
AdamW (class in torchium.optimizers)
AdaptiveLogSoftmaxWithLoss (class in torchium.losses)
AlphaIoULoss (class in torchium.losses)
AngularMetricLoss (class in torchium.losses)
Apollo (class in torchium.optimizers)
ArcFaceMetricLoss (class in torchium.losses)
ASAM (class in torchium.optimizers)
B
BCELoss (class in torchium.losses)
BCEWithLogitsLoss (class in torchium.losses)
BERTScoreLoss (in module torchium.losses)
BetaTCVAELoss (class in torchium.losses)
BetaVAELoss (class in torchium.losses)
blank (torchium.losses.CTCLoss attribute)
BLEULoss (class in torchium.losses)
BoundaryLoss (class in torchium.losses)
C
CAGradLoss (class in torchium.losses)
centralize_gradient() (torchium.optimizers.GradientCentralization method)
CIoULoss (class in torchium.losses)
ClassBalancedLoss (class in torchium.losses)
CMAES (class in torchium.optimizers)
compute_sdf() (torchium.losses.BoundaryLoss method)
ContentLoss (class in torchium.losses)
ContrastiveLoss (class in torchium.losses)
ContrastiveMetricLoss (class in torchium.losses)
CosFaceMetricLoss (class in torchium.losses)
CosineEmbeddingLoss (class in torchium.losses)
CRFLoss (class in torchium.losses)
CrossEntropyLoss (class in torchium.losses)
CTCLoss (class in torchium.losses)
cutoffs (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
D
DDIMLoss (class in torchium.losses)
DDPMLoss (class in torchium.losses)
DiceLoss (class in torchium.losses)
,
[1]
DifferentialEvolution (class in torchium.optimizers)
DIoULoss (class in torchium.losses)
div_value (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
DTWLoss (class in torchium.losses)
DynamicLossBalancing (class in torchium.losses)
E
EIoULoss (class in torchium.losses)
ELBOLoss (class in torchium.losses)
eps (torchium.losses.GaussianNLLLoss attribute)
(torchium.losses.PoissonNLLLoss attribute)
(torchium.losses.TripletMarginLoss attribute)
F
FactorVAELoss (in module torchium.losses)
FastTextLoss (in module torchium.losses)
first_step() (torchium.optimizers.ASAM method)
(torchium.optimizers.GSAM method)
(torchium.optimizers.SAM method)
(torchium.optimizers.WSAM method)
FocalDetectionLoss (class in torchium.losses)
FocalLoss (class in torchium.losses)
FocalTverskyLoss (class in torchium.losses)
forward() (torchium.losses.AdaptiveLogSoftmaxWithLoss method)
(torchium.losses.AlphaIoULoss method)
(torchium.losses.AngularMetricLoss method)
(torchium.losses.ArcFaceMetricLoss method)
(torchium.losses.BCELoss method)
(torchium.losses.BCEWithLogitsLoss method)
(torchium.losses.BetaTCVAELoss method)
(torchium.losses.BLEULoss method)
(torchium.losses.BoundaryLoss method)
(torchium.losses.CAGradLoss method)
(torchium.losses.CIoULoss method)
(torchium.losses.ClassBalancedLoss method)
(torchium.losses.ContentLoss method)
(torchium.losses.ContrastiveLoss method)
(torchium.losses.ContrastiveMetricLoss method)
(torchium.losses.CosFaceMetricLoss method)
(torchium.losses.CosineEmbeddingLoss method)
(torchium.losses.CRFLoss method)
(torchium.losses.CTCLoss method)
(torchium.losses.DDPMLoss method)
(torchium.losses.DiceLoss method)
,
[1]
(torchium.losses.DIoULoss method)
(torchium.losses.DTWLoss method)
(torchium.losses.DynamicLossBalancing method)
(torchium.losses.EIoULoss method)
(torchium.losses.ELBOLoss method)
(torchium.losses.FocalDetectionLoss method)
(torchium.losses.FocalLoss method)
(torchium.losses.FocalTverskyLoss method)
(torchium.losses.GANLoss method)
(torchium.losses.GaussianNLLLoss method)
(torchium.losses.GIoULoss method)
(torchium.losses.GradNormLoss method)
(torchium.losses.HingeEmbeddingLoss method)
(torchium.losses.HingeGANLoss method)
(torchium.losses.IoULoss method)
(torchium.losses.KLDivLoss method)
(torchium.losses.L1Loss method)
(torchium.losses.LabelSmoothingLoss method)
(torchium.losses.LeastSquaresGANLoss method)
(torchium.losses.LogCoshLoss method)
(torchium.losses.LovaszLoss method)
(torchium.losses.LPIPSLoss method)
(torchium.losses.MarginRankingLoss method)
(torchium.losses.MSSSIMLoss method)
(torchium.losses.MultiLabelMarginLoss method)
(torchium.losses.MultiLabelSoftMarginLoss method)
(torchium.losses.MultiMarginLoss method)
(torchium.losses.MultiTaskLoss method)
(torchium.losses.NLLLoss method)
(torchium.losses.NPairLoss method)
(torchium.losses.PCGradLoss method)
(torchium.losses.PerceptualLoss method)
(torchium.losses.PerplexityLoss method)
(torchium.losses.PoissonNLLLoss method)
(torchium.losses.ProxyAnchorLoss method)
(torchium.losses.ProxyNCALoss method)
(torchium.losses.QuadrupletLoss method)
(torchium.losses.QuantileLoss method)
(torchium.losses.RelativistGANLoss method)
(torchium.losses.ScoreMatchingLoss method)
(torchium.losses.SoftMarginLoss method)
(torchium.losses.SpectralLoss method)
(torchium.losses.SphereFaceLoss method)
(torchium.losses.SSIMLoss method)
(torchium.losses.StructuredPredictionLoss method)
(torchium.losses.StyleLoss method)
(torchium.losses.TotalVariationLoss method)
(torchium.losses.TripletMarginLoss method)
(torchium.losses.TripletMarginWithDistanceLoss method)
(torchium.losses.TripletMetricLoss method)
(torchium.losses.TverskyLoss method)
,
[1]
(torchium.losses.UncertaintyWeightingLoss method)
(torchium.losses.VGGLoss method)
(torchium.losses.WassersteinLoss method)
(torchium.losses.Word2VecLoss method)
FTRL (class in torchium.optimizers)
full (torchium.losses.GaussianNLLLoss attribute)
(torchium.losses.PoissonNLLLoss attribute)
G
GANLoss (class in torchium.losses)
GaussianNLLLoss (class in torchium.losses)
GeneticAlgorithm (class in torchium.optimizers)
get_trace() (torchium.optimizers.AdaHessian method)
,
[1]
GIoULoss (class in torchium.losses)
GloVeLoss (in module torchium.losses)
GradientCentralization (class in torchium.optimizers)
GradNorm (class in torchium.optimizers)
GradNormLoss (class in torchium.losses)
GSAM (class in torchium.optimizers)
H
head (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
head_bias (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
HeavyBall (class in torchium.optimizers)
HingeEmbeddingLoss (class in torchium.losses)
HingeGANLoss (class in torchium.losses)
HuberLoss (class in torchium.losses)
I
ignore_index (torchium.losses.NLLLoss attribute)
in_features (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
IoULoss (class in torchium.losses)
K
KFAC (class in torchium.optimizers)
KLDivLoss (class in torchium.losses)
L
L1Loss (class in torchium.losses)
LabelSmoothingLoss (class in torchium.losses)
LAMB (class in torchium.optimizers)
LARS (class in torchium.optimizers)
LBFGS (class in torchium.optimizers)
LeastSquaresGANLoss (class in torchium.losses)
Lion (class in torchium.optimizers)
log_input (torchium.losses.PoissonNLLLoss attribute)
log_prob() (torchium.losses.AdaptiveLogSoftmaxWithLoss method)
LogCoshLoss (class in torchium.losses)
lookahead_step() (torchium.optimizers.LookSAM method)
LookSAM (class in torchium.optimizers)
lovasz_grad() (torchium.losses.LovaszLoss method)
lovasz_softmax_flat() (torchium.losses.LovaszLoss method)
LovaszLoss (class in torchium.losses)
LPIPSLoss (class in torchium.losses)
M
MADGRAD (class in torchium.optimizers)
MAELoss (class in torchium.losses)
margin (torchium.losses.CosineEmbeddingLoss attribute)
(torchium.losses.HingeEmbeddingLoss attribute)
(torchium.losses.MarginRankingLoss attribute)
(torchium.losses.MultiMarginLoss attribute)
(torchium.losses.TripletMarginLoss attribute)
(torchium.losses.TripletMarginWithDistanceLoss attribute)
MarginRankingLoss (class in torchium.losses)
METEORLoss (in module torchium.losses)
MSELoss (class in torchium.losses)
MSSSIMLoss (class in torchium.losses)
MultiLabelMarginLoss (class in torchium.losses)
MultiLabelSoftMarginLoss (class in torchium.losses)
MultiMarginLoss (class in torchium.losses)
MultiTaskLoss (class in torchium.losses)
N
n_classes (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
NAdam (class in torchium.optimizers)
NaturalGradient (class in torchium.optimizers)
NLLLoss (class in torchium.losses)
NovoGrad (class in torchium.optimizers)
NPairLoss (class in torchium.losses)
P
p (torchium.losses.MultiMarginLoss attribute)
(torchium.losses.TripletMarginLoss attribute)
ParticleSwarmOptimization (class in torchium.optimizers)
PCGrad (class in torchium.optimizers)
PCGradLoss (class in torchium.losses)
PerceptualLoss (class in torchium.losses)
PerplexityLoss (class in torchium.losses)
PoissonNLLLoss (class in torchium.losses)
predict() (torchium.losses.AdaptiveLogSoftmaxWithLoss method)
project_conflicting_gradients() (torchium.optimizers.PCGrad method)
ProxyAnchorLoss (class in torchium.losses)
ProxyNCALoss (class in torchium.losses)
Q
QuadrupletLoss (class in torchium.losses)
QuantileLoss (class in torchium.losses)
QuantumAnnealing (class in torchium.optimizers)
R
RAdam (class in torchium.optimizers)
Ranger (class in torchium.optimizers)
RelativistGANLoss (class in torchium.losses)
reset_parameters() (torchium.losses.AdaptiveLogSoftmaxWithLoss method)
RMSprop (class in torchium.optimizers)
ROUGELoss (in module torchium.losses)
Rprop (class in torchium.optimizers)
S
SAM (class in torchium.optimizers)
ScoreMatchingLoss (class in torchium.losses)
second_step() (torchium.optimizers.ASAM method)
(torchium.optimizers.GSAM method)
(torchium.optimizers.SAM method)
(torchium.optimizers.WSAM method)
SGD (class in torchium.optimizers)
Shampoo (class in torchium.optimizers)
SM3 (class in torchium.optimizers)
SmoothL1Loss (class in torchium.losses)
SoftMarginLoss (class in torchium.losses)
SparseAdam (class in torchium.optimizers)
SpectralLoss (class in torchium.losses)
SphereFaceLoss (class in torchium.losses)
SSIMLoss (class in torchium.losses)
step() (torchium.optimizers.AdaBelief method)
(torchium.optimizers.AdaBound method)
(torchium.optimizers.AdaFactor method)
(torchium.optimizers.AdaHessian method)
,
[1]
(torchium.optimizers.AdamP method)
(torchium.optimizers.Apollo method)
(torchium.optimizers.ASAM method)
(torchium.optimizers.CMAES method)
(torchium.optimizers.DifferentialEvolution method)
(torchium.optimizers.FTRL method)
(torchium.optimizers.GeneticAlgorithm method)
(torchium.optimizers.GradientCentralization method)
(torchium.optimizers.GradNorm method)
(torchium.optimizers.GSAM method)
(torchium.optimizers.HeavyBall method)
(torchium.optimizers.KFAC method)
(torchium.optimizers.LAMB method)
(torchium.optimizers.LARS method)
(torchium.optimizers.LBFGS method)
(torchium.optimizers.Lion method)
(torchium.optimizers.LookSAM method)
(torchium.optimizers.MADGRAD method)
(torchium.optimizers.NAdam method)
(torchium.optimizers.NaturalGradient method)
(torchium.optimizers.NovoGrad method)
(torchium.optimizers.ParticleSwarmOptimization method)
(torchium.optimizers.PCGrad method)
(torchium.optimizers.QuantumAnnealing method)
(torchium.optimizers.RAdam method)
(torchium.optimizers.Ranger method)
(torchium.optimizers.Rprop method)
(torchium.optimizers.SAM method)
(torchium.optimizers.Shampoo method)
(torchium.optimizers.SM3 method)
(torchium.optimizers.WSAM method)
(torchium.optimizers.Yogi method)
StructuredPredictionLoss (class in torchium.losses)
StyleLoss (class in torchium.losses)
swap (torchium.losses.TripletMarginLoss attribute)
(torchium.losses.TripletMarginWithDistanceLoss attribute)
T
tail (torchium.losses.AdaptiveLogSoftmaxWithLoss attribute)
TotalVariationLoss (class in torchium.losses)
TripletLoss (class in torchium.losses)
TripletMarginLoss (class in torchium.losses)
TripletMarginWithDistanceLoss (class in torchium.losses)
TripletMetricLoss (class in torchium.losses)
TverskyLoss (class in torchium.losses)
,
[1]
U
UncertaintyWeightingLoss (class in torchium.losses)
V
VGGLoss (class in torchium.losses)
W
WassersteinLoss (class in torchium.losses)
Word2VecLoss (class in torchium.losses)
WSAM (class in torchium.optimizers)
Y
Yogi (class in torchium.optimizers)
Z
zero_infinity (torchium.losses.CTCLoss attribute)