Index

A

AdaBelief (class in holocron.optim)

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

B

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) cspdarknet53() (in module holocron.models) cspdarknet53_mish() (in module holocron.models)

D

darknet19() (in module holocron.models) darknet24() (in module holocron.models) darknet53() (in module holocron.models)

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

E

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

F

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

freeze_bn() (in module holocron.trainer) freeze_model() (in module holocron.trainer) FReLU (class in holocron.nn)

G

GlobalAvgPool2d (class in holocron.nn)

H

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

holocron.trainer module holocron.utils module

L

LabelSmoothingCrossEntropy (class in holocron.nn) 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) lr_find() (holocron.trainer.Trainer method) lr_finder() (in module holocron.utils) ls_cross_entropy() (in module holocron.nn.functional)

M

Mish (class in holocron.nn) mish() (in module holocron.nn.functional) mixup_collate() (in module holocron.utils.data) MixupLoss (class in holocron.nn) module holocron holocron.optim holocron.trainer 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)

O

OneCycleScheduler (class in holocron.optim.lr_scheduler)

P

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

pyconv_resnet50() (in module holocron.models) pyconvhg_resnet50() (in module holocron.models)

R

RAdam (class in holocron.optim) RaLars (class in holocron.optim) res2net50_26w_4s() (in module holocron.models) resnet101() (in module holocron.models) resnet152() (in module holocron.models) resnet18() (in module holocron.models) resnet34() (in module holocron.models) resnet50() (in module holocron.models)

resnet50d() (in module holocron.models) resnext101_32x8d() (in module holocron.models) resnext50_32x4d() (in module holocron.models) rexnet1_0x() (in module holocron.models) rexnet1_3x() (in module holocron.models) rexnet1_5x() (in module holocron.models) rexnet2_0x() (in module holocron.models) rexnet2_2x() (in module 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)

SiLU (class in holocron.nn) silu() (in module holocron.nn.functional) sknet152() (in module holocron.models) sknet50() (in module 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)

U

unet() (in module holocron.models) unet3p() (in module holocron.models)

unetp() (in module holocron.models) unetpp() (in module holocron.models)

Y

yolov1() (in module holocron.models)

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