├── .gitignore ├── README.md ├── deit ├── datasets.py ├── engine.py ├── losses.py └── super_deit.py ├── lvvit ├── __init__.py ├── data │ ├── __init__.py │ ├── dataset.py │ ├── label_transforms_factory.py │ ├── loader.py │ ├── mixup.py │ └── random_augment_label.py ├── loss │ ├── __init__.py │ └── cross_entropy.py ├── models │ ├── __init__.py │ ├── layers.py │ └── super_lvvit.py └── utils │ ├── __init__.py │ └── utils.py ├── main_deit.py ├── main_lvvit.py ├── samplers.py ├── supervit.png ├── timm ├── __init__.py ├── data │ ├── __init__.py │ ├── auto_augment.py │ ├── config.py │ ├── constants.py │ ├── dataset.py │ ├── dataset_factory.py │ ├── distributed_sampler.py │ ├── loader.py │ ├── mixup.py │ ├── parsers │ │ ├── __init__.py │ │ ├── class_map.py │ │ ├── constants.py │ │ ├── parser.py │ │ ├── parser_factory.py │ │ ├── parser_image_folder.py │ │ ├── parser_image_in_tar.py │ │ ├── parser_image_tar.py │ │ └── parser_tfds.py │ ├── random_erasing.py │ ├── real_labels.py │ ├── tf_preprocessing.py │ ├── transforms.py │ └── transforms_factory.py ├── loss │ ├── __init__.py │ ├── asymmetric_loss.py │ ├── cross_entropy.py │ └── jsd.py ├── models │ ├── __init__.py │ ├── byobnet.py │ ├── cspnet.py │ ├── densenet.py │ ├── dla.py │ ├── dpn.py │ ├── efficientnet.py │ ├── efficientnet_blocks.py │ ├── efficientnet_builder.py │ ├── factory.py │ ├── features.py │ ├── gluon_resnet.py │ ├── gluon_xception.py │ ├── helpers.py │ ├── hrnet.py │ ├── inception_resnet_v2.py │ ├── inception_v3.py │ ├── inception_v4.py │ ├── layers │ │ ├── __init__.py │ │ ├── activations.py │ │ ├── activations_jit.py │ │ ├── activations_me.py │ │ ├── adaptive_avgmax_pool.py │ │ ├── anti_aliasing.py │ │ ├── blur_pool.py │ │ ├── cbam.py │ │ ├── classifier.py │ │ ├── cond_conv2d.py │ │ ├── config.py │ │ ├── conv2d_same.py │ │ ├── conv_bn_act.py │ │ ├── create_act.py │ │ ├── create_attn.py │ │ ├── create_conv2d.py │ │ ├── create_norm_act.py │ │ ├── drop.py │ │ ├── eca.py │ │ ├── evo_norm.py │ │ ├── helpers.py │ │ ├── inplace_abn.py │ │ ├── linear.py │ │ ├── median_pool.py │ │ ├── mixed_conv2d.py │ │ ├── norm_act.py │ │ ├── padding.py │ │ ├── pool2d_same.py │ │ ├── se.py │ │ ├── selective_kernel.py │ │ ├── separable_conv.py │ │ ├── space_to_depth.py │ │ ├── split_attn.py │ │ ├── split_batchnorm.py │ │ ├── std_conv.py │ │ ├── test_time_pool.py │ │ └── weight_init.py │ ├── mobilenetv3.py │ ├── nasnet.py │ ├── nfnet.py │ ├── pnasnet.py │ ├── pruned │ │ ├── ecaresnet101d_pruned.txt │ │ ├── ecaresnet50d_pruned.txt │ │ ├── efficientnet_b1_pruned.txt │ │ ├── efficientnet_b2_pruned.txt │ │ └── efficientnet_b3_pruned.txt │ ├── registry.py │ ├── regnet.py │ ├── res2net.py │ ├── resnest.py │ ├── resnet.py │ ├── resnetv2.py │ ├── rexnet.py │ ├── selecsls.py │ ├── senet.py │ ├── sknet.py │ ├── tresnet.py │ ├── vgg.py │ ├── vision_transformer.py │ ├── vovnet.py │ ├── xception.py │ └── xception_aligned.py ├── optim │ ├── __init__.py │ ├── adafactor.py │ ├── adahessian.py │ ├── adamp.py │ ├── adamw.py │ ├── lookahead.py │ ├── nadam.py │ ├── novograd.py │ ├── nvnovograd.py │ ├── optim_factory.py │ ├── radam.py │ ├── rmsprop_tf.py │ └── sgdp.py ├── scheduler │ ├── __init__.py │ ├── cosine_lr.py │ ├── plateau_lr.py │ ├── scheduler.py │ ├── scheduler_factory.py │ ├── step_lr.py │ └── tanh_lr.py ├── utils │ ├── __init__.py │ ├── agc.py │ ├── checkpoint_saver.py │ ├── clip_grad.py │ ├── cuda.py │ ├── distributed.py │ ├── jit.py │ ├── log.py │ ├── metrics.py │ ├── misc.py │ ├── model.py │ ├── model_ema.py │ └── summary.py └── version.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/README.md -------------------------------------------------------------------------------- /deit/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/deit/datasets.py -------------------------------------------------------------------------------- /deit/engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/deit/engine.py -------------------------------------------------------------------------------- /deit/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/deit/losses.py -------------------------------------------------------------------------------- /deit/super_deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/deit/super_deit.py -------------------------------------------------------------------------------- /lvvit/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /lvvit/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/data/__init__.py -------------------------------------------------------------------------------- /lvvit/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/data/dataset.py -------------------------------------------------------------------------------- /lvvit/data/label_transforms_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/data/label_transforms_factory.py -------------------------------------------------------------------------------- /lvvit/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/data/loader.py -------------------------------------------------------------------------------- /lvvit/data/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/data/mixup.py -------------------------------------------------------------------------------- /lvvit/data/random_augment_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/data/random_augment_label.py -------------------------------------------------------------------------------- /lvvit/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/loss/__init__.py -------------------------------------------------------------------------------- /lvvit/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/loss/cross_entropy.py -------------------------------------------------------------------------------- /lvvit/models/__init__.py: -------------------------------------------------------------------------------- 1 | from lvvit import * 2 | -------------------------------------------------------------------------------- /lvvit/models/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/models/layers.py -------------------------------------------------------------------------------- /lvvit/models/super_lvvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/models/super_lvvit.py -------------------------------------------------------------------------------- /lvvit/utils/__init__.py: -------------------------------------------------------------------------------- 1 | from .utils import load_pretrained_weights -------------------------------------------------------------------------------- /lvvit/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/lvvit/utils/utils.py -------------------------------------------------------------------------------- /main_deit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/main_deit.py -------------------------------------------------------------------------------- /main_lvvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/main_lvvit.py -------------------------------------------------------------------------------- /samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/samplers.py -------------------------------------------------------------------------------- /supervit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/supervit.png -------------------------------------------------------------------------------- /timm/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/__init__.py -------------------------------------------------------------------------------- /timm/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/__init__.py -------------------------------------------------------------------------------- /timm/data/auto_augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/auto_augment.py -------------------------------------------------------------------------------- /timm/data/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/config.py -------------------------------------------------------------------------------- /timm/data/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/constants.py -------------------------------------------------------------------------------- /timm/data/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/dataset.py -------------------------------------------------------------------------------- /timm/data/dataset_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/dataset_factory.py -------------------------------------------------------------------------------- /timm/data/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/distributed_sampler.py -------------------------------------------------------------------------------- /timm/data/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/loader.py -------------------------------------------------------------------------------- /timm/data/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/mixup.py -------------------------------------------------------------------------------- /timm/data/parsers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/__init__.py -------------------------------------------------------------------------------- /timm/data/parsers/class_map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/class_map.py -------------------------------------------------------------------------------- /timm/data/parsers/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/constants.py -------------------------------------------------------------------------------- /timm/data/parsers/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/parser.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/parser_factory.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/parser_image_folder.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_image_in_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/parser_image_in_tar.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_image_tar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/parser_image_tar.py -------------------------------------------------------------------------------- /timm/data/parsers/parser_tfds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/parsers/parser_tfds.py -------------------------------------------------------------------------------- /timm/data/random_erasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/random_erasing.py -------------------------------------------------------------------------------- /timm/data/real_labels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/real_labels.py -------------------------------------------------------------------------------- /timm/data/tf_preprocessing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/tf_preprocessing.py -------------------------------------------------------------------------------- /timm/data/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/transforms.py -------------------------------------------------------------------------------- /timm/data/transforms_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/data/transforms_factory.py -------------------------------------------------------------------------------- /timm/loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/loss/__init__.py -------------------------------------------------------------------------------- /timm/loss/asymmetric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/loss/asymmetric_loss.py -------------------------------------------------------------------------------- /timm/loss/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/loss/cross_entropy.py -------------------------------------------------------------------------------- /timm/loss/jsd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/loss/jsd.py -------------------------------------------------------------------------------- /timm/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/__init__.py -------------------------------------------------------------------------------- /timm/models/byobnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/byobnet.py -------------------------------------------------------------------------------- /timm/models/cspnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/cspnet.py -------------------------------------------------------------------------------- /timm/models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/densenet.py -------------------------------------------------------------------------------- /timm/models/dla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/dla.py -------------------------------------------------------------------------------- /timm/models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/dpn.py -------------------------------------------------------------------------------- /timm/models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/efficientnet.py -------------------------------------------------------------------------------- /timm/models/efficientnet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/efficientnet_blocks.py -------------------------------------------------------------------------------- /timm/models/efficientnet_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/efficientnet_builder.py -------------------------------------------------------------------------------- /timm/models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/factory.py -------------------------------------------------------------------------------- /timm/models/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/features.py -------------------------------------------------------------------------------- /timm/models/gluon_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/gluon_resnet.py -------------------------------------------------------------------------------- /timm/models/gluon_xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/gluon_xception.py -------------------------------------------------------------------------------- /timm/models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/helpers.py -------------------------------------------------------------------------------- /timm/models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/hrnet.py -------------------------------------------------------------------------------- /timm/models/inception_resnet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/inception_resnet_v2.py -------------------------------------------------------------------------------- /timm/models/inception_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/inception_v3.py -------------------------------------------------------------------------------- /timm/models/inception_v4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/inception_v4.py -------------------------------------------------------------------------------- /timm/models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/__init__.py -------------------------------------------------------------------------------- /timm/models/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/activations.py -------------------------------------------------------------------------------- /timm/models/layers/activations_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/activations_jit.py -------------------------------------------------------------------------------- /timm/models/layers/activations_me.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/activations_me.py -------------------------------------------------------------------------------- /timm/models/layers/adaptive_avgmax_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/adaptive_avgmax_pool.py -------------------------------------------------------------------------------- /timm/models/layers/anti_aliasing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/anti_aliasing.py -------------------------------------------------------------------------------- /timm/models/layers/blur_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/blur_pool.py -------------------------------------------------------------------------------- /timm/models/layers/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/cbam.py -------------------------------------------------------------------------------- /timm/models/layers/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/classifier.py -------------------------------------------------------------------------------- /timm/models/layers/cond_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/cond_conv2d.py -------------------------------------------------------------------------------- /timm/models/layers/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/config.py -------------------------------------------------------------------------------- /timm/models/layers/conv2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/conv2d_same.py -------------------------------------------------------------------------------- /timm/models/layers/conv_bn_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/conv_bn_act.py -------------------------------------------------------------------------------- /timm/models/layers/create_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/create_act.py -------------------------------------------------------------------------------- /timm/models/layers/create_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/create_attn.py -------------------------------------------------------------------------------- /timm/models/layers/create_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/create_conv2d.py -------------------------------------------------------------------------------- /timm/models/layers/create_norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/create_norm_act.py -------------------------------------------------------------------------------- /timm/models/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/drop.py -------------------------------------------------------------------------------- /timm/models/layers/eca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/eca.py -------------------------------------------------------------------------------- /timm/models/layers/evo_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/evo_norm.py -------------------------------------------------------------------------------- /timm/models/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/helpers.py -------------------------------------------------------------------------------- /timm/models/layers/inplace_abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/inplace_abn.py -------------------------------------------------------------------------------- /timm/models/layers/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/linear.py -------------------------------------------------------------------------------- /timm/models/layers/median_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/median_pool.py -------------------------------------------------------------------------------- /timm/models/layers/mixed_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/mixed_conv2d.py -------------------------------------------------------------------------------- /timm/models/layers/norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/norm_act.py -------------------------------------------------------------------------------- /timm/models/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/padding.py -------------------------------------------------------------------------------- /timm/models/layers/pool2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/pool2d_same.py -------------------------------------------------------------------------------- /timm/models/layers/se.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/se.py -------------------------------------------------------------------------------- /timm/models/layers/selective_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/selective_kernel.py -------------------------------------------------------------------------------- /timm/models/layers/separable_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/separable_conv.py -------------------------------------------------------------------------------- /timm/models/layers/space_to_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/space_to_depth.py -------------------------------------------------------------------------------- /timm/models/layers/split_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/split_attn.py -------------------------------------------------------------------------------- /timm/models/layers/split_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/split_batchnorm.py -------------------------------------------------------------------------------- /timm/models/layers/std_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/std_conv.py -------------------------------------------------------------------------------- /timm/models/layers/test_time_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/test_time_pool.py -------------------------------------------------------------------------------- /timm/models/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/layers/weight_init.py -------------------------------------------------------------------------------- /timm/models/mobilenetv3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/mobilenetv3.py -------------------------------------------------------------------------------- /timm/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/nasnet.py -------------------------------------------------------------------------------- /timm/models/nfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/nfnet.py -------------------------------------------------------------------------------- /timm/models/pnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/pnasnet.py -------------------------------------------------------------------------------- /timm/models/pruned/ecaresnet101d_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/pruned/ecaresnet101d_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/ecaresnet50d_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/pruned/ecaresnet50d_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/efficientnet_b1_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/pruned/efficientnet_b1_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/efficientnet_b2_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/pruned/efficientnet_b2_pruned.txt -------------------------------------------------------------------------------- /timm/models/pruned/efficientnet_b3_pruned.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/pruned/efficientnet_b3_pruned.txt -------------------------------------------------------------------------------- /timm/models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/registry.py -------------------------------------------------------------------------------- /timm/models/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/regnet.py -------------------------------------------------------------------------------- /timm/models/res2net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/res2net.py -------------------------------------------------------------------------------- /timm/models/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/resnest.py -------------------------------------------------------------------------------- /timm/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/resnet.py -------------------------------------------------------------------------------- /timm/models/resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/resnetv2.py -------------------------------------------------------------------------------- /timm/models/rexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/rexnet.py -------------------------------------------------------------------------------- /timm/models/selecsls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/selecsls.py -------------------------------------------------------------------------------- /timm/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/senet.py -------------------------------------------------------------------------------- /timm/models/sknet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/sknet.py -------------------------------------------------------------------------------- /timm/models/tresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/tresnet.py -------------------------------------------------------------------------------- /timm/models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/vgg.py -------------------------------------------------------------------------------- /timm/models/vision_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/vision_transformer.py -------------------------------------------------------------------------------- /timm/models/vovnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/vovnet.py -------------------------------------------------------------------------------- /timm/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/xception.py -------------------------------------------------------------------------------- /timm/models/xception_aligned.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/models/xception_aligned.py -------------------------------------------------------------------------------- /timm/optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/__init__.py -------------------------------------------------------------------------------- /timm/optim/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/adafactor.py -------------------------------------------------------------------------------- /timm/optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/adahessian.py -------------------------------------------------------------------------------- /timm/optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/adamp.py -------------------------------------------------------------------------------- /timm/optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/adamw.py -------------------------------------------------------------------------------- /timm/optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/lookahead.py -------------------------------------------------------------------------------- /timm/optim/nadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/nadam.py -------------------------------------------------------------------------------- /timm/optim/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/novograd.py -------------------------------------------------------------------------------- /timm/optim/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/nvnovograd.py -------------------------------------------------------------------------------- /timm/optim/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/optim_factory.py -------------------------------------------------------------------------------- /timm/optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/radam.py -------------------------------------------------------------------------------- /timm/optim/rmsprop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/rmsprop_tf.py -------------------------------------------------------------------------------- /timm/optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/optim/sgdp.py -------------------------------------------------------------------------------- /timm/scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/__init__.py -------------------------------------------------------------------------------- /timm/scheduler/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/cosine_lr.py -------------------------------------------------------------------------------- /timm/scheduler/plateau_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/plateau_lr.py -------------------------------------------------------------------------------- /timm/scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/scheduler.py -------------------------------------------------------------------------------- /timm/scheduler/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/scheduler_factory.py -------------------------------------------------------------------------------- /timm/scheduler/step_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/step_lr.py -------------------------------------------------------------------------------- /timm/scheduler/tanh_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/scheduler/tanh_lr.py -------------------------------------------------------------------------------- /timm/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/__init__.py -------------------------------------------------------------------------------- /timm/utils/agc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/agc.py -------------------------------------------------------------------------------- /timm/utils/checkpoint_saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/checkpoint_saver.py -------------------------------------------------------------------------------- /timm/utils/clip_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/clip_grad.py -------------------------------------------------------------------------------- /timm/utils/cuda.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/cuda.py -------------------------------------------------------------------------------- /timm/utils/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/distributed.py -------------------------------------------------------------------------------- /timm/utils/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/jit.py -------------------------------------------------------------------------------- /timm/utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/log.py -------------------------------------------------------------------------------- /timm/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/metrics.py -------------------------------------------------------------------------------- /timm/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/misc.py -------------------------------------------------------------------------------- /timm/utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/model.py -------------------------------------------------------------------------------- /timm/utils/model_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/model_ema.py -------------------------------------------------------------------------------- /timm/utils/summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/timm/utils/summary.py -------------------------------------------------------------------------------- /timm/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.4.5' 2 | -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lmbxmu/SuperViT/HEAD/utils.py --------------------------------------------------------------------------------