Index

A

AdaBelief (class in holocron.optim) AdamP (class in holocron.optim) Adan (class in holocron.optim)

Add2d (class in holocron.nn) add2d() (in module holocron.nn.functional) AdEMAMix (class in holocron.optim)

B

BinaryClassificationTrainer (class in holocron.trainer)

BlurPool2d (class in holocron.nn) box_giou() (in module holocron.ops)

C

check_setup() (holocron.trainer.Trainer method) ciou_loss() (in module holocron.ops) ClassBalancedWrapper (class in holocron.nn) ClassificationTrainer (class in holocron.trainer) complement_cross_entropy() (in module holocron.nn.functional) ComplementCrossEntropy (class in holocron.nn) concat_downsample2d() (in module holocron.nn.functional) ConcatDownsample2d (class in holocron.nn) convnext_atto() (in module holocron.models) ConvNeXt_Atto_Checkpoint (class in holocron.models) convnext_base() (in module holocron.models)

convnext_femto() (in module holocron.models) convnext_large() (in module holocron.models) convnext_nano() (in module holocron.models) convnext_pico() (in module holocron.models) convnext_small() (in module holocron.models) convnext_tiny() (in module holocron.models) convnext_xl() (in module holocron.models) cspdarknet53() (in module holocron.models) CSPDarknet53_Checkpoint (class in holocron.models) cspdarknet53_mish() (in module holocron.models) CSPDarknet53_Mish_Checkpoint (class in holocron.models)

D

darknet19() (in module holocron.models) Darknet19_Checkpoint (class in holocron.models) darknet24() (in module holocron.models) darknet53() (in module holocron.models) Darknet53_Checkpoint (class in holocron.models)

DetectionTrainer (class in holocron.trainer) dice_loss() (in module holocron.nn.functional) DiceLoss (class in holocron.nn) diou_loss() (in module holocron.ops) DropBlock2d (class in holocron.nn) dropblock2d() (in module holocron.nn.functional)

E

evaluate() (holocron.trainer.BinaryClassificationTrainer method) (holocron.trainer.ClassificationTrainer method) (holocron.trainer.DetectionTrainer method) (holocron.trainer.SegmentationTrainer method)

F

find_image_size() (in module holocron.utils) find_lr() (holocron.trainer.Trainer method) fit_n_epochs() (holocron.trainer.Trainer method) focal_loss() (in module holocron.nn.functional) FocalLoss (class in holocron.nn)

forward() (holocron.transforms.RandomZoomOut method) (holocron.transforms.Resize method) freeze_bn() (in module holocron.trainer) freeze_model() (in module holocron.trainer) FReLU (class in holocron.nn)

G

GlobalAvgPool2d (class in holocron.nn)

GlobalMaxPool2d (class in holocron.nn)

H

hard_mish() (in module holocron.nn.functional) HardMish (class in holocron.nn) holocron.optim module holocron.trainer module

holocron.transforms module holocron.utils module

I

Involution2d (class in holocron.nn)

L

LAMB (class in holocron.optim) LambdaLayer (class in holocron.nn)

LARS (class in holocron.optim) load() (holocron.trainer.Trainer method) Lookahead (class in holocron.optim.wrapper)

M

Mixup() (in module holocron.utils.data) mobileone_s0() (in module holocron.models) MobileOne_S0_Checkpoint (class in holocron.models) mobileone_s1() (in module holocron.models) MobileOne_S1_Checkpoint (class in holocron.models) mobileone_s2() (in module holocron.models) MobileOne_S2_Checkpoint (class in holocron.models) mobileone_s3() (in module holocron.models)

MobileOne_S3_Checkpoint (class in holocron.models) module holocron.optim holocron.trainer holocron.transforms holocron.utils multilabel_cross_entropy() (in module holocron.nn.functional) MultiLabelCrossEntropy (class in holocron.nn) MutualChannelLoss (class in holocron.nn)

N

nl_relu() (in module holocron.nn.functional) NLReLU (class in holocron.nn)

norm_conv2d() (in module holocron.nn.functional) NormConv2d (class in holocron.nn)

P

plot_recorder() (holocron.trainer.Trainer method) PolyLoss (class in holocron.nn)

PyConv2d (class in holocron.nn) pyconv_resnet50() (in module holocron.models) pyconvhg_resnet50() (in module holocron.models)

R

RaLars (class in holocron.optim) RandomZoomOut (class in holocron.transforms) repvgg_a0() (in module holocron.models) RepVGG_A0_Checkpoint (class in holocron.models) repvgg_a1() (in module holocron.models) RepVGG_A1_Checkpoint (class in holocron.models) repvgg_a2() (in module holocron.models) RepVGG_A2_Checkpoint (class in holocron.models) repvgg_b0() (in module holocron.models) RepVGG_B0_Checkpoint (class in holocron.models) repvgg_b1() (in module holocron.models) RepVGG_B1_Checkpoint (class in holocron.models) repvgg_b2() (in module holocron.models) RepVGG_B2_Checkpoint (class in holocron.models) repvgg_b3() (in module holocron.models) res2net50_26w_4s() (in module holocron.models) Res2Net50_26w_4s_Checkpoint (class in holocron.models) Resize (class in holocron.transforms) resnet101() (in module holocron.models) resnet152() (in module holocron.models)

resnet18() (in module holocron.models) ResNet18_Checkpoint (class in holocron.models) resnet34() (in module holocron.models) ResNet34_Checkpoint (class in holocron.models) resnet50() (in module holocron.models) ResNet50_Checkpoint (class in holocron.models) resnet50d() (in module holocron.models) ResNet50D_Checkpoint (class in holocron.models) resnext101_32x8d() (in module holocron.models) resnext50_32x4d() (in module holocron.models) ResNeXt50_32x4d_Checkpoint (class in holocron.models) rexnet1_0x() (in module holocron.models) ReXNet1_0x_Checkpoint (class in holocron.models) rexnet1_3x() (in module holocron.models) ReXNet1_3x_Checkpoint (class in holocron.models) rexnet1_5x() (in module holocron.models) ReXNet1_5x_Checkpoint (class in holocron.models) rexnet2_0x() (in module holocron.models) ReXNet2_0x_Checkpoint (class in holocron.models) rexnet2_2x() (in module holocron.models) ReXNet2_2x_Checkpoint (class in holocron.models)

S

SAM (class in holocron.nn) save() (holocron.trainer.Trainer method) Scout (class in holocron.optim.wrapper) SegmentationTrainer (class in holocron.trainer) set_device() (holocron.trainer.Trainer method)

sknet101() (in module holocron.models) sknet152() (in module holocron.models) sknet50() (in module holocron.models) SKNet50_Checkpoint (class in holocron.models) SlimConv2d (class in holocron.nn) SPP (class in holocron.nn)

T

TAdam (class in holocron.optim) to_cuda() (holocron.trainer.Trainer method)

Trainer (class in holocron.trainer) tridentnet50() (in module holocron.models) TripletAttention (class in holocron.nn)

U

unet() (in module holocron.models.segmentation) unet2() (in module holocron.models.segmentation) unet3p() (in module holocron.models.segmentation) unet_rexnet13() (in module holocron.models.segmentation)

unet_tvresnet34() (in module holocron.models.segmentation) unet_tvvgg11() (in module holocron.models.segmentation) unetp() (in module holocron.models.segmentation) unetpp() (in module holocron.models.segmentation)

Y

yolov1() (in module holocron.models.detection)

yolov2() (in module holocron.models.detection) yolov4() (in module holocron.models.detection)

Z

z_pool() (in module holocron.nn.functional)

ZPool (class in holocron.nn)