basecls

• basecls.configs
  • BaseConfig
    • BaseConfig.link_log_dir()
  • EffNetConfig
  • EffNetLiteConfig
  • HRNetConfig
  • MBConfig
  • RegNetConfig
  • RepVGGConfig
  • ResMLPConfig
  • ResNetConfig
  • SNetConfig
  • SwinConfig
  • VGGConfig
  • ViTConfig
  • basecls.configs.base_cfg
    • BaseConfig
      • BaseConfig.link_log_dir()
  • basecls.configs.effnet_cfg
    • EffNetConfig
    • EffNetLiteConfig
  • basecls.configs.hrnet_cfg
    • HRNetConfig
  • basecls.configs.mbnet_cfg
    • MBConfig
  • basecls.configs.regnet_cfg
    • RegNetConfig
  • basecls.configs.repvgg_cfg
    • RepVGGConfig
  • basecls.configs.resmlp_cfg
    • ResMLPConfig
  • basecls.configs.resnet_cfg
    • ResNetConfig
  • basecls.configs.snet_cfg
    • SNetConfig
  • basecls.configs.swin_cfg
    • SwinConfig
  • basecls.configs.vgg_cfg
    • VGGConfig
  • basecls.configs.vit_cfg
    • ViTConfig
• basecls.data
  • FakeData
    • FakeData.build()
  • FolderLoader
    • FolderLoader.build()
  • build_dataloader()
  • build_dataset()
  • AutoAugment
    • AutoAugment.build()
  • ColorAugment
    • ColorAugment.build()
  • RandAugment
    • RandAugment.build()
  • SimpleAugment
    • SimpleAugment.build()
  • build_mixup()
  • build_transform()
  • basecls.data.augment
    • TorchAutoAugment
    • TorchRandAugment
  • basecls.data.build
    • FakeData
      • FakeData.build()
    • FolderLoader
      • FolderLoader.build()
  • basecls.data.const
  • basecls.data.dataloader
    • build_dataloader()
  • basecls.data.dataset
    • build_dataset()
  • basecls.data.fake_data
    • FakeDataLoader
  • basecls.data.mixup
    • MixupCutmixTransform
      • MixupCutmixTransform.apply_batch()
    • MixupCutmixCollator
      • MixupCutmixCollator.apply()
  • basecls.data.rand_erase
    • RandomErasing
      • RandomErasing.apply_batch()
  • basecls.data.transform
    • build_transform()
    • AutoAugment
      • AutoAugment.build()
    • SimpleAugment
      • SimpleAugment.build()
    • ColorAugment
      • ColorAugment.build()
    • RandAugment
      • RandAugment.build()
    • build_mixup()
• basecls.engine
  • DefaultHooks
    • DefaultHooks.build()
  • AccEvaluator
    • AccEvaluator.ResultType
    • AccEvaluator.preprocess()
    • AccEvaluator.postprocess()
    • AccEvaluator.evaluate()
  • ClsTester
    • ClsTester.test()
  • 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
    • DefaultHooks
      • DefaultHooks.build()
  • basecls.engine.evaluator
    • AccEvaluator
      • AccEvaluator.ResultType
      • AccEvaluator.preprocess()
      • AccEvaluator.postprocess()
      • AccEvaluator.evaluate()
  • basecls.engine.hooks
    • CheckpointHook
      • CheckpointHook.after_epoch()
      • CheckpointHook.after_train()
    • EvalHook
      • EvalHook.after_epoch()
      • EvalHook.after_train()
      • EvalHook.test()
    • LoggerHook
      • LoggerHook.before_train()
      • LoggerHook.after_train()
      • LoggerHook.before_iter()
      • LoggerHook.after_iter()
      • LoggerHook.get_loss_str()
      • LoggerHook.get_stat_str()
      • LoggerHook.get_memory_str()
      • LoggerHook.get_train_info_str()
      • LoggerHook.get_time_str()
    • LRSchedulerHook
      • LRSchedulerHook.before_epoch()
      • LRSchedulerHook.get_lr_factor()
      • LRSchedulerHook.total_lr
    • PreciseBNHook
      • PreciseBNHook.before_train()
      • PreciseBNHook.after_epoch()
    • ResumeHook
      • ResumeHook.before_train()
    • TensorboardHook
      • TensorboardHook.before_train()
      • TensorboardHook.after_train()
      • TensorboardHook.after_iter()
      • TensorboardHook.write()
      • TensorboardHook.calc_iter()
  • basecls.engine.tester
    • ClsTester
      • ClsTester.test()
  • 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
    • ELU.forward()
  • HSigmoid
    • HSigmoid.forward()
  • HSwish
    • HSwish.forward()
  • ReLU6
    • ReLU6.forward()
  • Tanh
    • Tanh.forward()
  • activation()
  • ClsHead
    • ClsHead.forward()
  • MBV3Head
    • MBV3Head.forward()
  • VGGHead
    • VGGHead.forward()
  • build_head()
  • BinaryCrossEntropy
    • BinaryCrossEntropy.forward()
  • CrossEntropy
    • CrossEntropy.forward()
  • build_loss()
  • SE
    • SE.avg_pool
    • SE.f_ex
    • SE.forward()
  • DropPath
    • DropPath.forward()
  • conv2d()
  • gap2d()
  • linear()
  • norm2d()
  • pool2d()
  • Preprocess
    • Preprocess.forward()
  • adjust_block_compatibility()
  • compute_precise_bn_stats()
  • init_vit_weights()
  • init_weights()
  • lecun_normal_()
  • make_divisible()
  • trunc_normal_()
  • basecls.layers.activations
    • activation()
    • ELU
      • ELU.forward()
    • HSigmoid
      • HSigmoid.forward()
    • HSwish
      • HSwish.forward()
    • ReLU6
      • ReLU6.forward()
    • Tanh
      • Tanh.forward()
  • basecls.layers.heads
    • build_head()
    • ClsHead
      • ClsHead.forward()
    • MBV3Head
      • MBV3Head.forward()
    • VGGHead
      • VGGHead.forward()
  • basecls.layers.losses
    • build_loss()
    • BinaryCrossEntropy
      • BinaryCrossEntropy.forward()
    • CrossEntropy
      • CrossEntropy.forward()
  • basecls.layers.modules
    • conv2d()
    • norm2d()
    • pool2d()
    • gap2d()
    • linear()
    • SE
      • SE.avg_pool
      • SE.f_ex
      • SE.forward()
    • DropPath
      • DropPath.forward()
  • basecls.layers.wrapper
    • Preprocess
      • Preprocess.forward()
    • adjust_block_compatibility()
    • calculate_fan_in_and_fan_out()
    • compute_precise_bn_stats()
    • init_weights()
    • init_vit_weights()
    • trunc_normal_()
    • lecun_normal_()
    • make_divisible()
• basecls.models
  • build_model()
  • load_model()
  • sync_model()
  • EffNet
    • EffNet.forward()
    • EffNet.get_block_func()
  • HRNet
    • HRNet.forward()
  • MBNet
    • MBNet.forward()
  • RegNet
  • RepVGG
    • RepVGG.forward()
    • RepVGG.convert_to_deploy()
  • ResMLP
    • ResMLP.forward()
  • ResNet
    • ResNet.forward()
    • ResNet.get_stem_func()
    • ResNet.get_block_func()
  • SNetV2
    • SNetV2.forward()
  • SwinTransformer
    • SwinTransformer.forward()
  • VGG
    • VGG.forward()
  • ViT
    • ViT.init_weights()
    • ViT.forward()
    • ViT.load_state_dict()
  • basecls.models.build
    • build_model()
    • load_model()
    • sync_model()
  • basecls.models.effnet
    • FuseMBConv
      • FuseMBConv.forward()
    • EffNet
      • EffNet.forward()
      • EffNet.get_block_func()
  • basecls.models.hrnet
    • UpsampleNearest
      • UpsampleNearest.forward()
    • HRFusion
      • HRFusion.forward()
    • HRModule
      • HRModule.forward()
    • HRTrans
      • HRTrans.forward()
    • HRStage
      • HRStage.forward()
    • HRMerge
      • HRMerge.forward()
    • HRNet
      • HRNet.forward()
  • basecls.models.mbnet
    • MBConv
      • MBConv.forward()
    • MBStage
      • MBStage.forward()
    • MBNet
      • MBNet.forward()
  • basecls.models.regnet
    • RegBottleneckBlock
      • RegBottleneckBlock.forward()
    • RegNet
  • basecls.models.repvgg
    • RepVGGBlock
      • RepVGGBlock.forward()
      • RepVGGBlock.convert_to_deploy()
    • RepVGG
      • RepVGG.forward()
      • RepVGG.convert_to_deploy()
  • basecls.models.resmlp
    • Affine
      • Affine.forward()
    • ResMLPBlock
      • ResMLPBlock.forward()
    • ResMLP
      • ResMLP.forward()
  • basecls.models.resnet
    • ResBasicBlock
      • ResBasicBlock.forward()
    • ResBottleneckBlock
      • ResBottleneckBlock.forward()
    • ResDeepStem
      • ResDeepStem.forward()
    • ResStem
      • ResStem.forward()
    • SimpleStem
      • SimpleStem.forward()
    • AnyStage
      • AnyStage.forward()
    • ResNet
      • ResNet.forward()
      • ResNet.get_stem_func()
      • ResNet.get_block_func()
  • basecls.models.snet
    • SNV2Block
      • SNV2Block.forward()
    • SNV2XceptionBlock
    • SNetV2
      • SNetV2.forward()
  • basecls.models.swin
    • window_partition()
    • window_reverse()
    • WindowAttention
      • WindowAttention.forward()
    • PatchMerging
      • PatchMerging.forward()
    • SwinBlock
      • SwinBlock.forward()
    • SwinBasicLayer
      • SwinBasicLayer.forward()
    • SwinTransformer
      • SwinTransformer.forward()
  • basecls.models.vgg
    • VGGStage
      • VGGStage.forward()
    • VGG
      • VGG.forward()
  • basecls.models.vit
    • PatchEmbed
      • PatchEmbed.forward()
    • Attention
      • Attention.forward()
    • FFN
      • FFN.forward()
    • EncoderBlock
      • EncoderBlock.forward()
    • ViT
      • ViT.init_weights()
      • ViT.forward()
      • ViT.load_state_dict()
• basecls.solver
  • BaseSolver
    • BaseSolver.build()
  • DefaultSolver
    • DefaultSolver.build()
    • DefaultSolver.build_optimizer()
  • Solver
    • Solver.grad_manager
    • Solver.grad_scaler
    • Solver.optimizer
  • basecls.solver.optimizer
    • LAMB
    • LARS
    • SGD
    • basecls.solver.optimizer.lamb
      • LAMB
    • basecls.solver.optimizer.lars
      • LARS
    • basecls.solver.optimizer.sgd
      • SGD
  • basecls.solver.build
    • Solver
      • Solver.grad_manager
      • Solver.grad_scaler
      • Solver.optimizer
    • BaseSolver
      • BaseSolver.build()
    • DefaultSolver
      • DefaultSolver.build()
      • DefaultSolver.build_optimizer()
  • basecls.solver.weight_decay
    • get_param_groups()
• basecls.tools
  • basecls.tools.cls_test
    • make_parser()
    • worker()
    • build()
    • main()
  • basecls.tools.cls_train
    • default_parser()
    • worker()
    • build()
    • main()
• basecls.utils
  • Registry
    • Registry.register()
    • Registry.get()
    • Registry.items()
    • Registry.keys()
    • Registry.values()
  • set_nccl_env()
  • set_num_threads()
  • default_logging()
  • setup_logger()
  • registers
    • registers.augments
    • registers.dataloaders
    • registers.hooks
    • registers.models
    • registers.solvers
    • registers.trainers
  • recursive_update()
  • basecls.utils.env
    • set_nccl_env()
    • set_num_threads()
  • basecls.utils.logger
    • default_logging()
    • setup_logger()
• basecls.zoo
  • basecls.zoo.benchmark
    • main()
    • worker()
    • ClsBench
      • ClsBench.benchmark()
      • ClsBench.model_step()
    • TrainBench
      • TrainBench.model_step()
    • EvalBench
      • EvalBench.model_step()
  • basecls.zoo.benchmark_timm
  • basecls.zoo.testing_all
    • make_parser()
    • worker()
    • build()
    • main()
  • basecls.zoo.utils
    • Meta
      • Meta.uid
      • Meta.series
      • Meta.name
      • Meta.create_time
      • Meta.basecls_version
      • Meta.flops
      • Meta.params
      • Meta.activations
      • Meta.img_size
      • Meta.acc1
      • Meta.acc5
    • get_series_and_name_and_id()
    • purify_weight()