├── CODE_OF_CONDUCT.MD ├── CONTRIBUTING.MD ├── LICENSE ├── README.md ├── configs ├── efflite0.yaml ├── mbv2-140-A.yaml ├── mbv2-140-B.yaml ├── mbv2-140-C.yaml ├── mbv2-140-D.yaml ├── mbv2-140-E.yaml ├── mbv2-140-F.yaml ├── mbv2-140.yaml └── mbv2.yaml ├── data ├── __init__.py ├── build.py ├── cached_image_folder.py ├── samplers.py └── zipreader.py ├── images ├── framework.png ├── overview.gif └── results.png ├── logger.py ├── loss_ops.py ├── lr_scheduler.py ├── main.py ├── misc ├── collect_log.py ├── config.py ├── config_ds.py ├── ds_utils.py ├── logger.py ├── loss_ops.py ├── lr_scheduler.py ├── optimizer.py └── utils.py ├── models ├── __init__.py ├── build.py ├── efficientnet.py ├── efficientnet_blocks.py ├── efficientnet_builder.py ├── efficientnet_ds.py ├── factory.py ├── features.py ├── helpers.py ├── hub.py ├── layers │ ├── __init__.py │ ├── activations.py │ ├── activations_jit.py │ ├── activations_me.py │ ├── adaptive_avgmax_pool.py │ ├── attention_pool2d.py │ ├── blur_pool.py │ ├── bottleneck_attn.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 │ ├── gather_excite.py │ ├── global_context.py │ ├── halo_attn.py │ ├── helpers.py │ ├── inplace_abn.py │ ├── lambda_layer.py │ ├── linear.py │ ├── median_pool.py │ ├── mixed_conv2d.py │ ├── mlp.py │ ├── non_local_attn.py │ ├── norm.py │ ├── norm_act.py │ ├── padding.py │ ├── patch_embed.py │ ├── pool2d_same.py │ ├── selective_kernel.py │ ├── separable_conv.py │ ├── space_to_depth.py │ ├── split_attn.py │ ├── split_batchnorm.py │ ├── squeeze_excite.py │ ├── std_conv.py │ ├── test_time_pool.py │ └── weight_init.py ├── registry.py └── utils_ds.py ├── optimizer.py └── utils.py /CODE_OF_CONDUCT.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/CODE_OF_CONDUCT.MD -------------------------------------------------------------------------------- /CONTRIBUTING.MD: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/CONTRIBUTING.MD -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/README.md -------------------------------------------------------------------------------- /configs/efflite0.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/efflite0.yaml -------------------------------------------------------------------------------- /configs/mbv2-140-A.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140-A.yaml -------------------------------------------------------------------------------- /configs/mbv2-140-B.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140-B.yaml -------------------------------------------------------------------------------- /configs/mbv2-140-C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140-C.yaml -------------------------------------------------------------------------------- /configs/mbv2-140-D.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140-D.yaml -------------------------------------------------------------------------------- /configs/mbv2-140-E.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140-E.yaml -------------------------------------------------------------------------------- /configs/mbv2-140-F.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140-F.yaml -------------------------------------------------------------------------------- /configs/mbv2-140.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2-140.yaml -------------------------------------------------------------------------------- /configs/mbv2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/configs/mbv2.yaml -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/data/__init__.py -------------------------------------------------------------------------------- /data/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/data/build.py -------------------------------------------------------------------------------- /data/cached_image_folder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/data/cached_image_folder.py -------------------------------------------------------------------------------- /data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/data/samplers.py -------------------------------------------------------------------------------- /data/zipreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/data/zipreader.py -------------------------------------------------------------------------------- /images/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/images/framework.png -------------------------------------------------------------------------------- /images/overview.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/images/overview.gif -------------------------------------------------------------------------------- /images/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/images/results.png -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/logger.py -------------------------------------------------------------------------------- /loss_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/loss_ops.py -------------------------------------------------------------------------------- /lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/lr_scheduler.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/main.py -------------------------------------------------------------------------------- /misc/collect_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/collect_log.py -------------------------------------------------------------------------------- /misc/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/config.py -------------------------------------------------------------------------------- /misc/config_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/config_ds.py -------------------------------------------------------------------------------- /misc/ds_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/ds_utils.py -------------------------------------------------------------------------------- /misc/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/logger.py -------------------------------------------------------------------------------- /misc/loss_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/loss_ops.py -------------------------------------------------------------------------------- /misc/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/lr_scheduler.py -------------------------------------------------------------------------------- /misc/optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/optimizer.py -------------------------------------------------------------------------------- /misc/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/misc/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/build.py -------------------------------------------------------------------------------- /models/efficientnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/efficientnet.py -------------------------------------------------------------------------------- /models/efficientnet_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/efficientnet_blocks.py -------------------------------------------------------------------------------- /models/efficientnet_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/efficientnet_builder.py -------------------------------------------------------------------------------- /models/efficientnet_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/efficientnet_ds.py -------------------------------------------------------------------------------- /models/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/factory.py -------------------------------------------------------------------------------- /models/features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/features.py -------------------------------------------------------------------------------- /models/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/helpers.py -------------------------------------------------------------------------------- /models/hub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/hub.py -------------------------------------------------------------------------------- /models/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/__init__.py -------------------------------------------------------------------------------- /models/layers/activations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/activations.py -------------------------------------------------------------------------------- /models/layers/activations_jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/activations_jit.py -------------------------------------------------------------------------------- /models/layers/activations_me.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/activations_me.py -------------------------------------------------------------------------------- /models/layers/adaptive_avgmax_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/adaptive_avgmax_pool.py -------------------------------------------------------------------------------- /models/layers/attention_pool2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/attention_pool2d.py -------------------------------------------------------------------------------- /models/layers/blur_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/blur_pool.py -------------------------------------------------------------------------------- /models/layers/bottleneck_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/bottleneck_attn.py -------------------------------------------------------------------------------- /models/layers/cbam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/cbam.py -------------------------------------------------------------------------------- /models/layers/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/classifier.py -------------------------------------------------------------------------------- /models/layers/cond_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/cond_conv2d.py -------------------------------------------------------------------------------- /models/layers/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/config.py -------------------------------------------------------------------------------- /models/layers/conv2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/conv2d_same.py -------------------------------------------------------------------------------- /models/layers/conv_bn_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/conv_bn_act.py -------------------------------------------------------------------------------- /models/layers/create_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/create_act.py -------------------------------------------------------------------------------- /models/layers/create_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/create_attn.py -------------------------------------------------------------------------------- /models/layers/create_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/create_conv2d.py -------------------------------------------------------------------------------- /models/layers/create_norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/create_norm_act.py -------------------------------------------------------------------------------- /models/layers/drop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/drop.py -------------------------------------------------------------------------------- /models/layers/eca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/eca.py -------------------------------------------------------------------------------- /models/layers/evo_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/evo_norm.py -------------------------------------------------------------------------------- /models/layers/gather_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/gather_excite.py -------------------------------------------------------------------------------- /models/layers/global_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/global_context.py -------------------------------------------------------------------------------- /models/layers/halo_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/halo_attn.py -------------------------------------------------------------------------------- /models/layers/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/helpers.py -------------------------------------------------------------------------------- /models/layers/inplace_abn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/inplace_abn.py -------------------------------------------------------------------------------- /models/layers/lambda_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/lambda_layer.py -------------------------------------------------------------------------------- /models/layers/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/linear.py -------------------------------------------------------------------------------- /models/layers/median_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/median_pool.py -------------------------------------------------------------------------------- /models/layers/mixed_conv2d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/mixed_conv2d.py -------------------------------------------------------------------------------- /models/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/mlp.py -------------------------------------------------------------------------------- /models/layers/non_local_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/non_local_attn.py -------------------------------------------------------------------------------- /models/layers/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/norm.py -------------------------------------------------------------------------------- /models/layers/norm_act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/norm_act.py -------------------------------------------------------------------------------- /models/layers/padding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/padding.py -------------------------------------------------------------------------------- /models/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/patch_embed.py -------------------------------------------------------------------------------- /models/layers/pool2d_same.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/pool2d_same.py -------------------------------------------------------------------------------- /models/layers/selective_kernel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/selective_kernel.py -------------------------------------------------------------------------------- /models/layers/separable_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/separable_conv.py -------------------------------------------------------------------------------- /models/layers/space_to_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/space_to_depth.py -------------------------------------------------------------------------------- /models/layers/split_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/split_attn.py -------------------------------------------------------------------------------- /models/layers/split_batchnorm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/split_batchnorm.py -------------------------------------------------------------------------------- /models/layers/squeeze_excite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/squeeze_excite.py -------------------------------------------------------------------------------- /models/layers/std_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/std_conv.py -------------------------------------------------------------------------------- /models/layers/test_time_pool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/test_time_pool.py -------------------------------------------------------------------------------- /models/layers/weight_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/layers/weight_init.py -------------------------------------------------------------------------------- /models/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/registry.py -------------------------------------------------------------------------------- /models/utils_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/models/utils_ds.py -------------------------------------------------------------------------------- /optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/optimizer.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/facebookresearch/DepthShrinker/HEAD/utils.py --------------------------------------------------------------------------------