basecls#
- basecls.configs
BaseConfig
EffNetConfig
EffNetLiteConfig
HRNetConfig
MBConfig
RegNetConfig
RepVGGConfig
ResMLPConfig
ResNetConfig
SNetConfig
SwinConfig
VGGConfig
ViTConfig
- basecls.configs.base_cfg
- basecls.configs.effnet_cfg
- basecls.configs.hrnet_cfg
- basecls.configs.mbnet_cfg
- basecls.configs.regnet_cfg
- basecls.configs.repvgg_cfg
- basecls.configs.resmlp_cfg
- basecls.configs.resnet_cfg
- basecls.configs.snet_cfg
- basecls.configs.swin_cfg
- basecls.configs.vgg_cfg
- basecls.configs.vit_cfg
- basecls.data
FakeData
FolderLoader
build_dataloader()
build_dataset()
AutoAugment
ColorAugment
RandAugment
SimpleAugment
build_mixup()
build_transform()
- basecls.data.augment
- basecls.data.build
- basecls.data.const
- basecls.data.dataloader
- basecls.data.dataset
- basecls.data.fake_data
- basecls.data.mixup
- basecls.data.rand_erase
- basecls.data.transform
- basecls.engine
DefaultHooks
AccEvaluator
ClsTester
ClsTrainer
ClsTrainer.cfg
ClsTrainer.model
ClsTrainer.ema
ClsTrainer.dataloader
ClsTrainer.solver
ClsTrainer.progress
ClsTrainer.loss
ClsTrainer.meter
ClsTrainer.train()
ClsTrainer.before_train()
ClsTrainer.before_epoch()
ClsTrainer.after_epoch()
ClsTrainer.train_one_iter()
ClsTrainer.model_step()
ClsTrainer.modify_grad()
ClsTrainer.model_ema_step()
- basecls.engine.build
- basecls.engine.evaluator
- basecls.engine.hooks
- basecls.engine.tester
- basecls.engine.trainer
ClsTrainer
ClsTrainer.cfg
ClsTrainer.model
ClsTrainer.ema
ClsTrainer.dataloader
ClsTrainer.solver
ClsTrainer.progress
ClsTrainer.loss
ClsTrainer.meter
ClsTrainer.train()
ClsTrainer.before_train()
ClsTrainer.before_epoch()
ClsTrainer.after_epoch()
ClsTrainer.train_one_iter()
ClsTrainer.model_step()
ClsTrainer.modify_grad()
ClsTrainer.model_ema_step()
- basecls.layers
ELU
HSigmoid
HSwish
ReLU6
Tanh
activation()
ClsHead
MBV3Head
VGGHead
build_head()
BinaryCrossEntropy
CrossEntropy
build_loss()
SE
DropPath
conv2d()
gap2d()
linear()
norm2d()
pool2d()
Preprocess
adjust_block_compatibility()
compute_precise_bn_stats()
init_vit_weights()
init_weights()
lecun_normal_()
make_divisible()
trunc_normal_()
- basecls.layers.activations
- basecls.layers.heads
- basecls.layers.losses
- basecls.layers.modules
- basecls.layers.wrapper
- basecls.models
build_model()
load_model()
sync_model()
EffNet
HRNet
MBNet
RegNet
RepVGG
ResMLP
ResNet
SNetV2
SwinTransformer
VGG
ViT
- basecls.models.build
- basecls.models.effnet
- basecls.models.hrnet
- basecls.models.mbnet
- basecls.models.regnet
- basecls.models.repvgg
- basecls.models.resmlp
- basecls.models.resnet
- basecls.models.snet
- basecls.models.swin
- basecls.models.vgg
- basecls.models.vit
- basecls.solver
- basecls.tools
- basecls.utils
- basecls.zoo