├── .gitignore ├── LICENSE ├── README.md ├── configs ├── models │ ├── DARTS │ │ ├── DARTS_V2_cifar.yaml │ │ └── DARTS_V2_imagenet.yaml │ ├── GreedyNAS │ │ ├── GreedyNAS-A.yaml │ │ ├── GreedyNAS-B.yaml │ │ └── GreedyNAS-C.yaml │ ├── MCT-NAS │ │ ├── MCT-NAS-A.yaml │ │ ├── MCT-NAS-B.yaml │ │ └── MCT-NAS-C.yaml │ ├── MobileNetV2 │ │ ├── MobileNetV2.yaml │ │ └── MobileNetV2_cifar10.yaml │ ├── PC-DARTS │ │ └── PC-DARTS_imagenet.yaml │ ├── ProxylessNAS │ │ └── ProxylessR-mobile.yaml │ ├── ResNet │ │ ├── resnet-50.yaml │ │ ├── resnext-50_32x4d.yaml │ │ ├── seresnet-50.yaml │ │ └── seresnext-50_32x4d.yaml │ └── VGG │ │ ├── vgg16_cifar10.yaml │ │ └── vgg16_cifar100.yaml └── strategies │ ├── CIFAR │ └── cifar.yaml │ ├── DARTS │ ├── darts.yaml │ ├── darts_cifar10.yaml │ └── pc-darts.yaml │ ├── DyRep │ ├── cifar.yaml │ ├── mbv1.yaml │ ├── repvgg_baseline.yaml │ ├── repvgg_strong.yaml │ └── resnet.yaml │ ├── MBV2 │ ├── mbv2.yaml │ ├── mbv2_cifar10.yaml │ └── mbv2_se_aa.yaml │ ├── deit │ └── deit_tiny.yaml │ ├── distill │ ├── diffkd │ │ ├── diffkd_b1.yaml │ │ ├── diffkd_b2.yaml │ │ └── diffkd_b3.yaml │ ├── dist_b2.yaml │ ├── dist_cifar.yaml │ ├── resnet_dist.yaml │ └── resnet_kdt4.yaml │ ├── lightvit │ └── config.yaml │ ├── local_mamba │ └── config.yaml │ └── resnet │ ├── resnet.yaml │ └── seresnext.yaml ├── lib ├── __init__.py ├── dataset │ ├── __init__.py │ ├── augment_ops.py │ ├── builder.py │ ├── dataloader.py │ ├── dataset.py │ ├── file_io.py │ ├── mixup.py │ └── transform.py ├── models │ ├── __init__.py │ ├── builder.py │ ├── cifar │ │ ├── ShuffleNetv1.py │ │ ├── ShuffleNetv2.py │ │ ├── __init__.py │ │ ├── classifier.py │ │ ├── mobilenetv2.py │ │ ├── resnet.py │ │ ├── resnetv2.py │ │ ├── util.py │ │ ├── vgg.py │ │ └── wrn.py │ ├── darts_model.py │ ├── lightvit.py │ ├── local_vim.py │ ├── local_vmamba.py │ ├── losses │ │ ├── __init__.py │ │ ├── cross_entropy.py │ │ ├── diffkd │ │ │ ├── __init__.py │ │ │ ├── diffkd.py │ │ │ ├── diffkd_modules.py │ │ │ └── scheduling_ddim.py │ │ ├── dist_kd.py │ │ ├── kd_loss.py │ │ └── kl_div.py │ ├── mamba │ │ ├── __init__.py │ │ ├── local_scan.py │ │ ├── multi_mamba.py │ │ └── rope.py │ ├── mdconv.py │ ├── mobilenet_v1.py │ ├── nas_model.py │ ├── operations.py │ ├── operations_resnet.py │ ├── resnet.py │ └── utils │ │ ├── __init__.py │ │ ├── dbb │ │ ├── __init__.py │ │ ├── dbb_block.py │ │ └── dbb_transforms.py │ │ ├── dbb_converter.py │ │ ├── dyrep.py │ │ └── recal_bn.py └── utils │ ├── args.py │ ├── dist_utils.py │ ├── gen_network.py │ ├── measure.py │ ├── misc.py │ ├── model_ema.py │ ├── optim.py │ └── scheduler.py └── tools ├── convert.py ├── convert_ckpt.py ├── dist_run.sh ├── dist_train.sh ├── slurm_run.sh ├── slurm_train.sh ├── speed_test.py ├── test.py ├── train.py └── vis_search_prob.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/README.md -------------------------------------------------------------------------------- /configs/models/DARTS/DARTS_V2_cifar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/DARTS/DARTS_V2_cifar.yaml -------------------------------------------------------------------------------- /configs/models/DARTS/DARTS_V2_imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/DARTS/DARTS_V2_imagenet.yaml -------------------------------------------------------------------------------- /configs/models/GreedyNAS/GreedyNAS-A.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/GreedyNAS/GreedyNAS-A.yaml -------------------------------------------------------------------------------- /configs/models/GreedyNAS/GreedyNAS-B.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/GreedyNAS/GreedyNAS-B.yaml -------------------------------------------------------------------------------- /configs/models/GreedyNAS/GreedyNAS-C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/GreedyNAS/GreedyNAS-C.yaml -------------------------------------------------------------------------------- /configs/models/MCT-NAS/MCT-NAS-A.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/MCT-NAS/MCT-NAS-A.yaml -------------------------------------------------------------------------------- /configs/models/MCT-NAS/MCT-NAS-B.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/MCT-NAS/MCT-NAS-B.yaml -------------------------------------------------------------------------------- /configs/models/MCT-NAS/MCT-NAS-C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/MCT-NAS/MCT-NAS-C.yaml -------------------------------------------------------------------------------- /configs/models/MobileNetV2/MobileNetV2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/MobileNetV2/MobileNetV2.yaml -------------------------------------------------------------------------------- /configs/models/MobileNetV2/MobileNetV2_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/MobileNetV2/MobileNetV2_cifar10.yaml -------------------------------------------------------------------------------- /configs/models/PC-DARTS/PC-DARTS_imagenet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/PC-DARTS/PC-DARTS_imagenet.yaml -------------------------------------------------------------------------------- /configs/models/ProxylessNAS/ProxylessR-mobile.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/ProxylessNAS/ProxylessR-mobile.yaml -------------------------------------------------------------------------------- /configs/models/ResNet/resnet-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/ResNet/resnet-50.yaml -------------------------------------------------------------------------------- /configs/models/ResNet/resnext-50_32x4d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/ResNet/resnext-50_32x4d.yaml -------------------------------------------------------------------------------- /configs/models/ResNet/seresnet-50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/ResNet/seresnet-50.yaml -------------------------------------------------------------------------------- /configs/models/ResNet/seresnext-50_32x4d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/ResNet/seresnext-50_32x4d.yaml -------------------------------------------------------------------------------- /configs/models/VGG/vgg16_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/VGG/vgg16_cifar10.yaml -------------------------------------------------------------------------------- /configs/models/VGG/vgg16_cifar100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/models/VGG/vgg16_cifar100.yaml -------------------------------------------------------------------------------- /configs/strategies/CIFAR/cifar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/CIFAR/cifar.yaml -------------------------------------------------------------------------------- /configs/strategies/DARTS/darts.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DARTS/darts.yaml -------------------------------------------------------------------------------- /configs/strategies/DARTS/darts_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DARTS/darts_cifar10.yaml -------------------------------------------------------------------------------- /configs/strategies/DARTS/pc-darts.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DARTS/pc-darts.yaml -------------------------------------------------------------------------------- /configs/strategies/DyRep/cifar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DyRep/cifar.yaml -------------------------------------------------------------------------------- /configs/strategies/DyRep/mbv1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DyRep/mbv1.yaml -------------------------------------------------------------------------------- /configs/strategies/DyRep/repvgg_baseline.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DyRep/repvgg_baseline.yaml -------------------------------------------------------------------------------- /configs/strategies/DyRep/repvgg_strong.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DyRep/repvgg_strong.yaml -------------------------------------------------------------------------------- /configs/strategies/DyRep/resnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/DyRep/resnet.yaml -------------------------------------------------------------------------------- /configs/strategies/MBV2/mbv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/MBV2/mbv2.yaml -------------------------------------------------------------------------------- /configs/strategies/MBV2/mbv2_cifar10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/MBV2/mbv2_cifar10.yaml -------------------------------------------------------------------------------- /configs/strategies/MBV2/mbv2_se_aa.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/MBV2/mbv2_se_aa.yaml -------------------------------------------------------------------------------- /configs/strategies/deit/deit_tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/deit/deit_tiny.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/diffkd/diffkd_b1.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/diffkd/diffkd_b1.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/diffkd/diffkd_b2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/diffkd/diffkd_b2.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/diffkd/diffkd_b3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/diffkd/diffkd_b3.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/dist_b2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/dist_b2.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/dist_cifar.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/dist_cifar.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/resnet_dist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/resnet_dist.yaml -------------------------------------------------------------------------------- /configs/strategies/distill/resnet_kdt4.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/distill/resnet_kdt4.yaml -------------------------------------------------------------------------------- /configs/strategies/lightvit/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/lightvit/config.yaml -------------------------------------------------------------------------------- /configs/strategies/local_mamba/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/local_mamba/config.yaml -------------------------------------------------------------------------------- /configs/strategies/resnet/resnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/resnet/resnet.yaml -------------------------------------------------------------------------------- /configs/strategies/resnet/seresnext.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/configs/strategies/resnet/seresnext.yaml -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/dataset/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/dataset/augment_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/augment_ops.py -------------------------------------------------------------------------------- /lib/dataset/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/builder.py -------------------------------------------------------------------------------- /lib/dataset/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/dataloader.py -------------------------------------------------------------------------------- /lib/dataset/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/dataset.py -------------------------------------------------------------------------------- /lib/dataset/file_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/file_io.py -------------------------------------------------------------------------------- /lib/dataset/mixup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/mixup.py -------------------------------------------------------------------------------- /lib/dataset/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/dataset/transform.py -------------------------------------------------------------------------------- /lib/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/__init__.py -------------------------------------------------------------------------------- /lib/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/builder.py -------------------------------------------------------------------------------- /lib/models/cifar/ShuffleNetv1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/ShuffleNetv1.py -------------------------------------------------------------------------------- /lib/models/cifar/ShuffleNetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/ShuffleNetv2.py -------------------------------------------------------------------------------- /lib/models/cifar/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/__init__.py -------------------------------------------------------------------------------- /lib/models/cifar/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/classifier.py -------------------------------------------------------------------------------- /lib/models/cifar/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/mobilenetv2.py -------------------------------------------------------------------------------- /lib/models/cifar/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/resnet.py -------------------------------------------------------------------------------- /lib/models/cifar/resnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/resnetv2.py -------------------------------------------------------------------------------- /lib/models/cifar/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/util.py -------------------------------------------------------------------------------- /lib/models/cifar/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/vgg.py -------------------------------------------------------------------------------- /lib/models/cifar/wrn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/cifar/wrn.py -------------------------------------------------------------------------------- /lib/models/darts_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/darts_model.py -------------------------------------------------------------------------------- /lib/models/lightvit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/lightvit.py -------------------------------------------------------------------------------- /lib/models/local_vim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/local_vim.py -------------------------------------------------------------------------------- /lib/models/local_vmamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/local_vmamba.py -------------------------------------------------------------------------------- /lib/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/__init__.py -------------------------------------------------------------------------------- /lib/models/losses/cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/cross_entropy.py -------------------------------------------------------------------------------- /lib/models/losses/diffkd/__init__.py: -------------------------------------------------------------------------------- 1 | from .diffkd import DiffKD 2 | -------------------------------------------------------------------------------- /lib/models/losses/diffkd/diffkd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/diffkd/diffkd.py -------------------------------------------------------------------------------- /lib/models/losses/diffkd/diffkd_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/diffkd/diffkd_modules.py -------------------------------------------------------------------------------- /lib/models/losses/diffkd/scheduling_ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/diffkd/scheduling_ddim.py -------------------------------------------------------------------------------- /lib/models/losses/dist_kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/dist_kd.py -------------------------------------------------------------------------------- /lib/models/losses/kd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/kd_loss.py -------------------------------------------------------------------------------- /lib/models/losses/kl_div.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/losses/kl_div.py -------------------------------------------------------------------------------- /lib/models/mamba/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/models/mamba/local_scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/mamba/local_scan.py -------------------------------------------------------------------------------- /lib/models/mamba/multi_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/mamba/multi_mamba.py -------------------------------------------------------------------------------- /lib/models/mamba/rope.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/mamba/rope.py -------------------------------------------------------------------------------- /lib/models/mdconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/mdconv.py -------------------------------------------------------------------------------- /lib/models/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/mobilenet_v1.py -------------------------------------------------------------------------------- /lib/models/nas_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/nas_model.py -------------------------------------------------------------------------------- /lib/models/operations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/operations.py -------------------------------------------------------------------------------- /lib/models/operations_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/operations_resnet.py -------------------------------------------------------------------------------- /lib/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/resnet.py -------------------------------------------------------------------------------- /lib/models/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/models/utils/dbb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/models/utils/dbb/dbb_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/utils/dbb/dbb_block.py -------------------------------------------------------------------------------- /lib/models/utils/dbb/dbb_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/utils/dbb/dbb_transforms.py -------------------------------------------------------------------------------- /lib/models/utils/dbb_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/utils/dbb_converter.py -------------------------------------------------------------------------------- /lib/models/utils/dyrep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/utils/dyrep.py -------------------------------------------------------------------------------- /lib/models/utils/recal_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/models/utils/recal_bn.py -------------------------------------------------------------------------------- /lib/utils/args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/args.py -------------------------------------------------------------------------------- /lib/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/dist_utils.py -------------------------------------------------------------------------------- /lib/utils/gen_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/gen_network.py -------------------------------------------------------------------------------- /lib/utils/measure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/measure.py -------------------------------------------------------------------------------- /lib/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/misc.py -------------------------------------------------------------------------------- /lib/utils/model_ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/model_ema.py -------------------------------------------------------------------------------- /lib/utils/optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/optim.py -------------------------------------------------------------------------------- /lib/utils/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/lib/utils/scheduler.py -------------------------------------------------------------------------------- /tools/convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/convert.py -------------------------------------------------------------------------------- /tools/convert_ckpt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/convert_ckpt.py -------------------------------------------------------------------------------- /tools/dist_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/dist_run.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/slurm_run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/slurm_run.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/speed_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/speed_test.py -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/train.py -------------------------------------------------------------------------------- /tools/vis_search_prob.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/hunto/image_classification_sota/HEAD/tools/vis_search_prob.py --------------------------------------------------------------------------------