gitextract_0n5u8q1i/ ├── .gitattributes ├── .github/ │ ├── FUNDING.yml │ ├── ISSUE_TEMPLATE/ │ │ ├── bug_report.md │ │ ├── config.yml │ │ └── feature_request.md │ └── workflows/ │ ├── build_documentation.yml │ ├── build_pr_documentation.yml │ ├── tests.yml │ ├── trufflehog.yml │ └── upload_pr_documentation.yml ├── .gitignore ├── CITATION.cff ├── CLAUDE.md ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── UPGRADING.md ├── avg_checkpoints.py ├── benchmark.py ├── bulk_runner.py ├── clean_checkpoint.py ├── convert/ │ ├── convert_from_mxnet.py │ └── convert_nest_flax.py ├── distributed_train.sh ├── hfdocs/ │ ├── README.md │ └── source/ │ ├── _toctree.yml │ ├── changes.mdx │ ├── feature_extraction.mdx │ ├── hf_hub.mdx │ ├── hparams.mdx │ ├── index.mdx │ ├── installation.mdx │ ├── models/ │ │ ├── adversarial-inception-v3.mdx │ │ ├── advprop.mdx │ │ ├── big-transfer.mdx │ │ ├── csp-darknet.mdx │ │ ├── csp-resnet.mdx │ │ ├── csp-resnext.mdx │ │ ├── densenet.mdx │ │ ├── dla.mdx │ │ ├── dpn.mdx │ │ ├── ecaresnet.mdx │ │ ├── efficientnet-pruned.mdx │ │ ├── efficientnet.mdx │ │ ├── ensemble-adversarial.mdx │ │ ├── ese-vovnet.mdx │ │ ├── fbnet.mdx │ │ ├── gloun-inception-v3.mdx │ │ ├── gloun-resnet.mdx │ │ ├── gloun-resnext.mdx │ │ ├── gloun-senet.mdx │ │ ├── gloun-seresnext.mdx │ │ ├── gloun-xception.mdx │ │ ├── hrnet.mdx │ │ ├── ig-resnext.mdx │ │ ├── inception-resnet-v2.mdx │ │ ├── inception-v3.mdx │ │ ├── inception-v4.mdx │ │ ├── legacy-se-resnet.mdx │ │ ├── legacy-se-resnext.mdx │ │ ├── legacy-senet.mdx │ │ ├── mixnet.mdx │ │ ├── mnasnet.mdx │ │ ├── mobilenet-v2.mdx │ │ ├── mobilenet-v3.mdx │ │ ├── nasnet.mdx │ │ ├── noisy-student.mdx │ │ ├── pnasnet.mdx │ │ ├── regnetx.mdx │ │ ├── regnety.mdx │ │ ├── res2net.mdx │ │ ├── res2next.mdx │ │ ├── resnest.mdx │ │ ├── resnet-d.mdx │ │ ├── resnet.mdx │ │ ├── resnext.mdx │ │ ├── rexnet.mdx │ │ ├── se-resnet.mdx │ │ ├── selecsls.mdx │ │ ├── seresnext.mdx │ │ ├── skresnet.mdx │ │ ├── skresnext.mdx │ │ ├── spnasnet.mdx │ │ ├── ssl-resnet.mdx │ │ ├── swsl-resnet.mdx │ │ ├── swsl-resnext.mdx │ │ ├── tf-efficientnet-condconv.mdx │ │ ├── tf-efficientnet-lite.mdx │ │ ├── tf-efficientnet.mdx │ │ ├── tf-inception-v3.mdx │ │ ├── tf-mixnet.mdx │ │ ├── tf-mobilenet-v3.mdx │ │ ├── tresnet.mdx │ │ ├── wide-resnet.mdx │ │ └── xception.mdx │ ├── models.mdx │ ├── quickstart.mdx │ ├── reference/ │ │ ├── data.mdx │ │ ├── models.mdx │ │ ├── optimizers.mdx │ │ └── schedulers.mdx │ ├── results.mdx │ └── training_script.mdx ├── hubconf.py ├── inference.py ├── onnx_export.py ├── onnx_validate.py ├── pyproject.toml ├── requirements-dev.txt ├── requirements.txt ├── results/ │ ├── README.md │ ├── benchmark-infer-amp-nchw-pt113-cu117-rtx3090.csv │ ├── benchmark-infer-amp-nchw-pt210-cu121-rtx3090.csv │ ├── benchmark-infer-amp-nchw-pt240-cu124-rtx3090.csv │ ├── benchmark-infer-amp-nchw-pt240-cu124-rtx4090-dynamo.csv │ ├── benchmark-infer-amp-nchw-pt240-cu124-rtx4090.csv │ ├── benchmark-infer-amp-nchw-pt291-cu128-4090-dynamo.csv │ ├── benchmark-infer-amp-nchw-pt291-cu128-4090.csv │ ├── benchmark-infer-amp-nchw-pt291-cu130-5090-dynamo.csv │ ├── benchmark-infer-amp-nchw-pt291-cu130-5090.csv │ ├── benchmark-infer-amp-nchw-pt291-cu130-pro6000maxq-dynamo.csv │ ├── benchmark-infer-amp-nchw-pt291-cu130-pro6000maxq.csv │ ├── benchmark-infer-amp-nhwc-pt113-cu117-rtx3090.csv │ ├── benchmark-infer-amp-nhwc-pt210-cu121-rtx3090.csv │ ├── benchmark-infer-amp-nhwc-pt240-cu124-rtx3090.csv │ ├── benchmark-infer-amp-nhwc-pt240-cu124-rtx4090.csv │ ├── benchmark-infer-amp_bf16-nchw-pt291-cu130-pro6000maxq-dynamo.csv │ ├── benchmark-infer-bf16-nchw-pt291-cu130-pro6000maxq-dynamo.csv │ ├── benchmark-infer-fp32-nchw-pt221-cpu-i9_10940x-dynamo.csv │ ├── benchmark-infer-fp32-nchw-pt240-cpu-i7_12700h-dynamo.csv │ ├── benchmark-infer-fp32-nchw-pt240-cpu-i9_10940x-dynamo.csv │ ├── benchmark-train-amp-nchw-pt112-cu113-rtx3090.csv │ ├── benchmark-train-amp-nhwc-pt112-cu113-rtx3090.csv │ ├── generate_csv_results.py │ ├── model_metadata-in1k.csv │ ├── results-imagenet-a-clean.csv │ ├── results-imagenet-a.csv │ ├── results-imagenet-r-clean.csv │ ├── results-imagenet-r.csv │ ├── results-imagenet-real.csv │ ├── results-imagenet.csv │ ├── results-imagenetv2-matched-frequency.csv │ └── results-sketch.csv ├── setup.cfg ├── tests/ │ ├── __init__.py │ ├── test_checkpoint_loading.py │ ├── test_layers.py │ ├── test_layers_drop.py │ ├── test_layers_pool.py │ ├── test_models.py │ ├── test_optim.py │ ├── test_scheduler.py │ └── test_utils.py ├── timm/ │ ├── __init__.py │ ├── data/ │ │ ├── __init__.py │ │ ├── _info/ │ │ │ ├── imagenet12k_synsets.txt │ │ │ ├── imagenet21k_goog_synsets.txt │ │ │ ├── imagenet21k_goog_to_12k_indices.txt │ │ │ ├── imagenet21k_goog_to_22k_indices.txt │ │ │ ├── imagenet21k_miil_synsets.txt │ │ │ ├── imagenet21k_miil_w21_synsets.txt │ │ │ ├── imagenet22k_ms_synsets.txt │ │ │ ├── imagenet22k_ms_to_12k_indices.txt │ │ │ ├── imagenet22k_ms_to_22k_indices.txt │ │ │ ├── imagenet22k_synsets.txt │ │ │ ├── imagenet22k_to_12k_indices.txt │ │ │ ├── imagenet_a_indices.txt │ │ │ ├── imagenet_a_synsets.txt │ │ │ ├── imagenet_r_indices.txt │ │ │ ├── imagenet_r_synsets.txt │ │ │ ├── imagenet_real_labels.json │ │ │ ├── imagenet_synset_to_definition.txt │ │ │ ├── imagenet_synset_to_lemma.txt │ │ │ ├── imagenet_synsets.txt │ │ │ ├── mini_imagenet_indices.txt │ │ │ └── mini_imagenet_synsets.txt │ │ ├── auto_augment.py │ │ ├── config.py │ │ ├── constants.py │ │ ├── dataset.py │ │ ├── dataset_factory.py │ │ ├── dataset_info.py │ │ ├── distributed_sampler.py │ │ ├── imagenet_info.py │ │ ├── loader.py │ │ ├── mixup.py │ │ ├── naflex_dataset.py │ │ ├── naflex_loader.py │ │ ├── naflex_mixup.py │ │ ├── naflex_random_erasing.py │ │ ├── naflex_transforms.py │ │ ├── random_erasing.py │ │ ├── readers/ │ │ │ ├── __init__.py │ │ │ ├── class_map.py │ │ │ ├── img_extensions.py │ │ │ ├── reader.py │ │ │ ├── reader_factory.py │ │ │ ├── reader_hfds.py │ │ │ ├── reader_hfids.py │ │ │ ├── reader_image_folder.py │ │ │ ├── reader_image_in_tar.py │ │ │ ├── reader_image_tar.py │ │ │ ├── reader_tfds.py │ │ │ ├── reader_wds.py │ │ │ └── shared_count.py │ │ ├── real_labels.py │ │ ├── tf_preprocessing.py │ │ ├── transforms.py │ │ └── transforms_factory.py │ ├── layers/ │ │ ├── __init__.py │ │ ├── _fx.py │ │ ├── activations.py │ │ ├── activations_me.py │ │ ├── adaptive_avgmax_pool.py │ │ ├── attention.py │ │ ├── attention2d.py │ │ ├── attention_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 │ │ ├── coord_attn.py │ │ ├── create_act.py │ │ ├── create_attn.py │ │ ├── create_conv2d.py │ │ ├── create_norm.py │ │ ├── create_norm_act.py │ │ ├── diff_attention.py │ │ ├── drop.py │ │ ├── eca.py │ │ ├── evo_norm.py │ │ ├── fast_norm.py │ │ ├── filter_response_norm.py │ │ ├── format.py │ │ ├── gather_excite.py │ │ ├── global_context.py │ │ ├── grid.py │ │ ├── grn.py │ │ ├── halo_attn.py │ │ ├── helpers.py │ │ ├── hybrid_embed.py │ │ ├── inplace_abn.py │ │ ├── interpolate.py │ │ ├── lambda_layer.py │ │ ├── layer_scale.py │ │ ├── linear.py │ │ ├── median_pool.py │ │ ├── mixed_conv2d.py │ │ ├── ml_decoder.py │ │ ├── mlp.py │ │ ├── non_local_attn.py │ │ ├── norm.py │ │ ├── norm_act.py │ │ ├── other_pool.py │ │ ├── padding.py │ │ ├── patch_dropout.py │ │ ├── patch_embed.py │ │ ├── pool1d.py │ │ ├── pool2d_same.py │ │ ├── pos_embed.py │ │ ├── pos_embed_rel.py │ │ ├── pos_embed_sincos.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 │ │ ├── trace_utils.py │ │ ├── typing.py │ │ └── weight_init.py │ ├── loss/ │ │ ├── __init__.py │ │ ├── asymmetric_loss.py │ │ ├── binary_cross_entropy.py │ │ ├── cross_entropy.py │ │ └── jsd.py │ ├── models/ │ │ ├── __init__.py │ │ ├── _builder.py │ │ ├── _efficientnet_blocks.py │ │ ├── _efficientnet_builder.py │ │ ├── _factory.py │ │ ├── _features.py │ │ ├── _features_fx.py │ │ ├── _helpers.py │ │ ├── _hub.py │ │ ├── _manipulate.py │ │ ├── _pretrained.py │ │ ├── _prune.py │ │ ├── _pruned/ │ │ │ ├── ecaresnet101d_pruned.txt │ │ │ ├── ecaresnet50d_pruned.txt │ │ │ ├── efficientnet_b1_pruned.txt │ │ │ ├── efficientnet_b2_pruned.txt │ │ │ └── efficientnet_b3_pruned.txt │ │ ├── _registry.py │ │ ├── beit.py │ │ ├── byoanet.py │ │ ├── byobnet.py │ │ ├── cait.py │ │ ├── coat.py │ │ ├── convit.py │ │ ├── convmixer.py │ │ ├── convnext.py │ │ ├── crossvit.py │ │ ├── csatv2.py │ │ ├── cspnet.py │ │ ├── davit.py │ │ ├── deit.py │ │ ├── densenet.py │ │ ├── dla.py │ │ ├── dpn.py │ │ ├── edgenext.py │ │ ├── efficientformer.py │ │ ├── efficientformer_v2.py │ │ ├── efficientnet.py │ │ ├── efficientvit_mit.py │ │ ├── efficientvit_msra.py │ │ ├── eva.py │ │ ├── factory.py │ │ ├── fasternet.py │ │ ├── fastvit.py │ │ ├── features.py │ │ ├── focalnet.py │ │ ├── fx_features.py │ │ ├── gcvit.py │ │ ├── ghostnet.py │ │ ├── hardcorenas.py │ │ ├── helpers.py │ │ ├── hgnet.py │ │ ├── hiera.py │ │ ├── hieradet_sam2.py │ │ ├── hrnet.py │ │ ├── hub.py │ │ ├── inception_next.py │ │ ├── inception_resnet_v2.py │ │ ├── inception_v3.py │ │ ├── inception_v4.py │ │ ├── layers/ │ │ │ └── __init__.py │ │ ├── levit.py │ │ ├── mambaout.py │ │ ├── maxxvit.py │ │ ├── metaformer.py │ │ ├── mlp_mixer.py │ │ ├── mobilenetv3.py │ │ ├── mobilenetv5.py │ │ ├── mobilevit.py │ │ ├── mvitv2.py │ │ ├── naflexvit.py │ │ ├── nasnet.py │ │ ├── nest.py │ │ ├── nextvit.py │ │ ├── nfnet.py │ │ ├── pit.py │ │ ├── pnasnet.py │ │ ├── pvt_v2.py │ │ ├── rdnet.py │ │ ├── registry.py │ │ ├── regnet.py │ │ ├── repghost.py │ │ ├── repvit.py │ │ ├── res2net.py │ │ ├── resnest.py │ │ ├── resnet.py │ │ ├── resnetv2.py │ │ ├── rexnet.py │ │ ├── selecsls.py │ │ ├── senet.py │ │ ├── sequencer.py │ │ ├── shvit.py │ │ ├── sknet.py │ │ ├── starnet.py │ │ ├── swiftformer.py │ │ ├── swin_transformer.py │ │ ├── swin_transformer_v2.py │ │ ├── swin_transformer_v2_cr.py │ │ ├── tiny_vit.py │ │ ├── tnt.py │ │ ├── tresnet.py │ │ ├── twins.py │ │ ├── vgg.py │ │ ├── visformer.py │ │ ├── vision_transformer.py │ │ ├── vision_transformer_hybrid.py │ │ ├── vision_transformer_relpos.py │ │ ├── vision_transformer_sam.py │ │ ├── vitamin.py │ │ ├── volo.py │ │ ├── vovnet.py │ │ ├── xception.py │ │ ├── xception_aligned.py │ │ └── xcit.py │ ├── optim/ │ │ ├── __init__.py │ │ ├── _optim_factory.py │ │ ├── _param_groups.py │ │ ├── _types.py │ │ ├── adabelief.py │ │ ├── adafactor.py │ │ ├── adafactor_bv.py │ │ ├── adahessian.py │ │ ├── adamp.py │ │ ├── adamw.py │ │ ├── adan.py │ │ ├── adopt.py │ │ ├── kron.py │ │ ├── lamb.py │ │ ├── laprop.py │ │ ├── lars.py │ │ ├── lion.py │ │ ├── lookahead.py │ │ ├── madgrad.py │ │ ├── mars.py │ │ ├── muon.py │ │ ├── nadam.py │ │ ├── nadamw.py │ │ ├── nvnovograd.py │ │ ├── optim_factory.py │ │ ├── radam.py │ │ ├── rmsprop_tf.py │ │ ├── sgdp.py │ │ └── sgdw.py │ ├── py.typed │ ├── scheduler/ │ │ ├── __init__.py │ │ ├── cosine_lr.py │ │ ├── multistep_lr.py │ │ ├── plateau_lr.py │ │ ├── poly_lr.py │ │ ├── scheduler.py │ │ ├── scheduler_factory.py │ │ ├── step_lr.py │ │ └── tanh_lr.py │ ├── task/ │ │ ├── __init__.py │ │ ├── classification.py │ │ ├── distillation.py │ │ ├── task.py │ │ └── token_distillation.py │ ├── utils/ │ │ ├── __init__.py │ │ ├── agc.py │ │ ├── attention_extract.py │ │ ├── checkpoint_saver.py │ │ ├── clip_grad.py │ │ ├── cuda.py │ │ ├── decay_batch.py │ │ ├── distributed.py │ │ ├── jit.py │ │ ├── log.py │ │ ├── metrics.py │ │ ├── misc.py │ │ ├── model.py │ │ ├── model_ema.py │ │ ├── onnx.py │ │ ├── random.py │ │ └── summary.py │ └── version.py ├── train.py └── validate.py