pytorch-image-models

mirrors/pytorch-image-models

Fork 0

mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00

Commit Graph

Select branches

Hide Pull Requests

0.5.x

0.6.x

adafactor_bv

aimv2

attn_update

augmentation_update

beit3_remap

beit3_remap_clean

bits_and_tpu

bulk_runner_tweaks

cache_dir

caojiaolong-main

cautious_optim

check_intermediate_feat

cleanup_vit_convert

cleanup_xla_model_fixes

clip_laion2b

convnext_and_copyright

convnext_nano_r384

convnext_shortcut

convnext_xxlarge_eps

convnext_zepto

convnextv2

convnormact_aa_none

dataset_info

device_amp_cleanup

dfnvit

diffattn

dinov2_reg4

dist_loss_tracking

dot_nine_cleanup

dynamic_resize_vit

edgenext_csp_and_more

efficientnet_attn

efficientnet_x

effnet_improve_features_only

eva

eva02

eva_pe_integration

fcossio-vit-maxpool

few_more_weights

fix_asymm_set_grad_enable

fix_hrnet_head

fix_imagenet22k_ms_mapping

fix_levit_load

fix_mnv4_query_strides

fix_onnx

fix_out_indices_order

fix_swin_pad

fix_tests

fix_tests_210

fix_window_trace

fixes-syncbn_pretrain_cfg_resolve

fixes_bce_regnet

florence2

focalnet_and_swin_refactor

freeze_norm_revisit

friday_weights

gh-pages

grodino-dataset_trust_remote

group_size_eff

half_prec_trainval

hf_hub_revisit

inception_next

inference_tests

intern300m

kron_flatten

kron_optimizer

levit_efficientformer_redux

local_dir

main

mambaout

mars_tweak

maxxvit_hub

mega_merge

mesa_ema

meshgrid_ndgrid

metaformer_baselines_for_vision

mnv4_first_weights

mobileclip_hub_id

mobileone

model_cleanup_docstrings_anno

more_attn

more_datasets

more_mobile

more_opt_stuff

more_tiny_test_models

more_vit

multi-weight

multi-weight_effnet_convnext

mw-deit

mw-final

mw-mlp_mixer

mw-more

mw-resnet

naflex

non_reentrant_ckpt

norm_mlp_classifier

norm_norm_norm

onnx_dynamo

onnx_export

openai_clip_resnet

openclip_weight_move

opt_mars_more

parse_kwargs

parse_kwargs2

patch_drop_refactor

perceiver

pos_embed_type_safe

preact_resnet18

pyproject_pdm

refactor-imports

refactor_pre_logits

resnetv2_18_34

samvit_fix_and_rope

sbb2_vit_hiera_weights

scaling_vit

script_cleanup

siglip2

siglip_update

sknet_kwargs_merge

small_384_weights

small_things

so150m2

swin_maxvit_resize

swin_v2

tiny_test_models

transforms_dataset_updates

update-test-pytorch

update_clip_pretrained

update_test_workflow

vit_and_bit_test_fixes

vit_packed

vit_relpos_refactor

vit_siglip_and_reg

warmup_prefix_fix

weights_only

wip-voidbag-accumulate-grad

xsepconv

yehuitang-Add-GhostNetV2

yuweihao-inception_next_atto

#1

#1007

#101

#1012

#1014

#1034

#1037

#105

#1055

#1056

#1056

#1058

#1061

#1063

#1063

#1069

#1083

#1091

#1094

#1105

#1112

#1113

#115

#1150

#117

#1187

#1190

#1194

#12

#1208

#1210

#1212

#1213

#1213

#122

#1222

#1229

#123

#1230

#1233

#1239

#1245

#1249

#125

#1251

#1253

#1256

#1259

#1265

#1266

#1270

#1271

#1273

#1274

#1276

#1277

#1294

#13

#1317

#132

#1322

#1327

#1330

#1336

#1340

#1345

#1349

#1351

#1354

#136

#1363

#1365

#137

#1381

#1383

#1385

#140

#141

#1414

#1415

#142

#1420

#143

#1439

#145

#146

#1467

#1471

#1473

#1476

#1479

#148

#1483

#1493

#1494

#1498

#150

#1500

#1502

#1520

#1522

#1525

#1527

#1532

#1538

#154

#1545

#1547

#155

#1552

#1553

#156

#1562

#1564

#1565

#1575

#1578

#1581

#1582

#1583

#1586

#1590

#1592

#1593

#16

#1606

#1612

#1614

#1617

#1621

#1622

#1624

#1627

#1628

#163

#1630

#1633

#1641

#1643

#1645

#1647

#1649

#1654

#1655

#1659

#1662

#1673

#1674

#1677

#1679

#1680

#1683

#1685

#1687

#1688

#1690

#1691

#1696

#1708

#1714

#1715

#1719

#1720

#1720

#1736

#1741

#1745

#1746

#1747

#1748

#175

#1756

#1760

#1761

#1766

#1778

#1784

#1785

#1789

#179

#1793

#1797

#1799

#1804

#1812

#1823

#183

#1830

#1834

#1835

#1837

#1841

#1846

#1847

#1850

#1852

#1866

#1873

#1876

#1888

#1889

#1890

#1894

#19

#1900

#1903

#191

#1911

#1913

#1914

#1916

#1918

#1919

#1920

#1923

#1925

#1926

#1930

#1937

#1945

#195

#1951

#1952

#1952

#1954

#1955

#1958

#1961

#1964

#1965

#1967

#1971

#1973

#1974

#1974

#1975

#1978

#1980

#1985

#1988

#199

#1993

#1995

#1996

#1998

#1998

#2

#2001

#2002

#2003

#2005

#2008

#2010

#2011

#2012

#2014

#2015

#2016

#2017

#2019

#2020

#2023

#2029

#2032

#2032

#2033

#2034

#2036

#2037

#2039

#2040

#2041

#2043

#2045

#2045

#2048

#2048

#2049

#2050

#2051

#2052

#2053

#2058

#2062

#2062

#2064

#2071

#2077

#2079

#208

#2082

#2089

#2090

#2091

#2092

#2096

#2098

#2102

#2105

#2108

#2109

#2111

#2113

#2117

#212

#2121

#2123

#2124

#2126

#213

#2136

#2138

#214

#2142

#2143

#2144

#2145

#2156

#216

#2162

#2164

#2165

#2167

#2167

#2168

#2169

#2174

#2175

#2176

#2176

#2178

#218

#2180

#2181

#2182

#2183

#2189

#2191

#2193

#2195

#2196

#2197

#2198

#2199

#2200

#2202

#2203

#2205

#2207

#2209

#2213

#2214

#2217

#2220

#2223

#2225

#2229

#2230

#2236

#2238

#2239

#2240

#2243

#2243

#2245

#2246

#2252

#2253

#2257

#2258

#2263

#2264

#2266

#227

#2274

#2275

#2277

#228

#2281

#2285

#2292

#2297

#2298

#2299

#23

#2304

#2305

#2308

#2311

#2312

#2314

#2314

#2316

#2317

#2319

#2319

#2320

#2321

#2322

#2325

#2326

#2328

#233

#2331

#2333

#2334

#2336

#2337

#2340

#2341

#2343

#2346

#2347

#2349

#2350

#2350

#2351

#2353

#2356

#2357

#2358

#2359

#2360

#2361

#2369

#237

#2372

#2372

#2376

#2379

#2381

#2382

#2384

#2386

#2386

#2387

#2388

#2389

#2394

#2396

#2397

#2398

#2399

#24

#240

#2400

#2402

#2403

#2404

#2405

#2406

#2408

#2409

#2412

#2413

#2415

#2416

#2417

#2418

#2418

#2421

#2423

#2426

#2426

#2427

#2429

#2431

#2433

#2439

#244

#2440

#2441

#2442

#2444

#2448

#2448

#2466

#2466

#2467

#2471

#2474

#2475

#2476

#2477

#2478

#2480

#2482

#2483

#2484

#2486

#2487

#2489

#2490

#2490

#2496

#2496

#2499

#250

#2501

#2501

#2502

#2503

#2504

#2504

#255

#256

#263

#268

#282

#284

#286

#29

#290

#297

#30

#300

#302

#31

#315

#32

#322

#323

#33

#334

#335

#35

#352

#367

#382

#389

#401

#413

#418

#419

#421

#425

#427

#429

#434

#437

#440

#447

#450

#459

#475

#489

#494

#497

#501

#502

#510

#52

#525

#528

#53

#533

#537

#542

#548

#550

#554

#555

#556

#561

#571

#580

#581

#595

#6

#603

#609

#612

#613

#62

#625

#626

#636

#637

#645

#647

#65

#651

#652

#653

#659

#66

#660

#668

#669

#679

#681

#685

#693

#7

#702

#714

#729

#731

#738

#74

#746

#747

#750

#768

#768

#771

#772

#800

#801

#804

#804

#805

#808

#812

#813

#816

#817

#82

#821

#828

#83

#841

#842

#853

#858

#87

#874

#875

#876

#88

#880

#882

#891

#898

#910

#913

#914

#915

#933

#94

#95

#964

#967

#968

#986

#989

#99

#993

v0.1-attn-weights

v0.1-cadene

v0.1-coat-weights

v0.1-dnf-weights

v0.1-effv2-weights

v0.1-ger-weights

v0.1-hrnet

v0.1-mvit-weights

v0.1-pit-weights

v0.1-regnet

v0.1-repvgg-weights

v0.1-res2net

v0.1-resnest

v0.1-rexnet

v0.1-rs-weights

v0.1-rsb-weights

v0.1-selecsls

v0.1-tpu-weights

v0.1-tresnet

v0.1-vitjx

v0.1-vt3p-weights

v0.1-weights

v0.1-weights-maxx

v0.1-weights-morevit

v0.1-weights-swinv2

v0.1.1

v0.2.1

v0.3.3

v0.3.4

v0.4.12

v0.4.5

v0.4.9

v0.5.4

v0.6.11

v0.6.12

v0.6.13

v0.6.5

v0.6.7

v0.8.10dev0

v0.8.13dev0

v0.8.17dev0

v0.8.2dev0

v0.8.6dev0

v0.9.0

v0.9.1

v0.9.10

v0.9.11

v0.9.12

v0.9.16

v0.9.2

v0.9.5

v0.9.6

v0.9.7

v0.9.8

v0.9.9

v1.0.10

v1.0.11

v1.0.12

v1.0.13

v1.0.14

v1.0.15

v1.0.3

v1.0.7

v1.0.8

v1.0.9

a22366e3ce

Merge pull request #2503 from huggingface/beit3_remap_clean main Ross Wightman 2025-05-30 16:40:28 -07:00
72858c193c Add siglip2 compatible naflex encoders. Add support to factorized pos embeds and 'aspect preserving mode' to Flex Embeds. Some more docstrings and typing. naflex Ross Wightman 2025-05-30 16:15:37 -07:00
211cf90721 Imports getting unwieldy in vision_transformer.py beit3_remap_clean Ross Wightman 2025-05-30 15:11:51 -07:00
1be7999993 Upload beit3 weights to hub, add pretrain weights Ross Wightman 2025-05-30 14:36:41 -07:00
bfa7ec917a Doing some Claude enabled docstring, type annotation and other cleanup model_cleanup_docstrings_anno Ross Wightman 2025-05-30 11:49:27 -07:00
3a3d98bc38 Fix parallel blocks missing scale args and vitamin MLP Ross Wightman 2025-05-29 11:34:19 -07:00
2ca94a6ce4 Compact _covert_beit3 fn Ross Wightman 2025-05-29 10:52:39 -07:00
38c5f3bc63 A cleaned up beit3 remap onto vision_transformer.py vit Ross Wightman 2025-05-29 10:33:09 -07:00
55e52c45ef Initial run through remapping beit3 -> vision_transformer.py beit3_remap Ross Wightman 2025-05-29 09:50:17 -07:00
a0a30a6b48

Update README.md Ross Wightman 2025-05-28 09:42:09 -07:00
3114a9f595

Merge pull request #2499 from brianhou0208/fast_model Ross Wightman 2025-05-26 18:49:51 -07:00
4209788233 Updated faster models w/ hub weight locations, commented out some checkpoint filter fns and minor renames Ross Wightman 2025-05-26 16:00:29 -07:00
b7ced7c40c torch.fx.wrap not working with older pytorch, trying register_notrace instead Ross Wightman 2025-05-25 14:13:36 -07:00
842a786626 A few more maybe_add_mask situations Ross Wightman 2025-05-25 08:51:56 -07:00
dd2c1418d0 Fix tracing of attention module with attn_mask support Ross Wightman 2025-05-24 21:13:01 -07:00
162f49295e Move naflex global pool into one fn that can be marked notrace Ross Wightman 2025-05-24 14:06:12 -07:00
2ad75e8023 Fix issue w/ MAP attention mask and no patch_valid Ross Wightman 2025-05-24 12:37:39 -07:00
d7d3538335 Add so400m model size for test, few tweaks. Ross Wightman 2025-05-23 18:29:30 -07:00
7bfe606d9f Merge remote-tracking branch 'origin/main' into naflex Ross Wightman 2025-05-23 09:41:49 -07:00
75823ab90e Merge branch 'main' into fast_model Ryan 2025-05-22 02:37:00 +08:00
e7925ea6d6 Add explicit disable of fc_norm to pe giant spatial variant Ross Wightman 2025-05-21 07:31:36 -07:00
f001b15ed3 NaFlex random erasing performance improvements, python loops were slow. Remove subregion mode, not going to be worth it. Ross Wightman 2025-05-20 17:03:46 -07:00
7624389fc9 Mixup cleanup, add prob support and train script integration. Add working loader based patch compatible RandomErasing for NaFlex mode. Ross Wightman 2025-05-20 14:38:03 -07:00
9790fea406 Merge branch 'main' into beit3 Ryan 2025-05-17 22:31:07 +08:00
bea1137583 Merge branch 'main' into fast_model Ryan 2025-05-17 22:25:34 +08:00
236b00cf2d add ghostnetv3 Ryan 2025-05-17 17:57:54 +08:00
edc37be1c3 Remove default 1000 classes from onnx export to use model default Ross Wightman 2025-05-15 14:14:53 -07:00
f14f6507ab

Merge pull request #2487 from huggingface/eva_pe_integration Ross Wightman 2025-05-15 12:56:03 -07:00
cabd26d018 Fix missing device arg in train.py, fix #2488 Ross Wightman 2025-05-14 13:51:43 -07:00
6b302f27a3

Merge pull request #2480 from brianhou0208/tnt Ross Wightman 2025-05-14 12:27:21 -07:00
1922ca5f1b fix fasternet Ryan 2025-05-15 02:28:17 +08:00
16d0b26e19 Fix torchscript issue with legacy tnt Ross Wightman 2025-05-14 09:33:41 -07:00
74ad32a67e Updated tnt model weights on hub, add back legacy model in case bwd compat Ross Wightman 2025-05-14 08:40:53 -07:00
cb98094792 Merge branch 'main' into fast_model Ryan 2025-05-14 08:30:33 +08:00
89d2952375 update group_matcher Ryan 2025-05-14 08:28:10 +08:00
b5a814e4c1 add giant model param Ryan 2025-05-12 00:24:15 +08:00
afe4375e77 update BEiT3 Ryan 2025-05-12 00:13:52 +08:00
008514934c Merge branch 'main' into beit3 Ryan 2025-05-11 22:56:10 +08:00
69b1fbcdc1 Merge branch 'main' into tnt Ryan 2025-05-11 22:45:48 +08:00
88b7ef6035 Disable dynamic_img_size default on PE models for now eva_pe_integration Ross Wightman 2025-05-10 15:00:29 -07:00
8fcbceb609 Add a WIP NaFlex compatible mixup/cutmix for testing Ross Wightman 2025-05-10 14:59:37 -07:00
fa7287faee Add EVA ViT based PE (Perceptual Encoder) impl Ross Wightman 2025-05-10 13:13:12 -07:00
81900a6bae remove download argument from torch_kwargs Ryan Ramos 2025-05-09 15:11:22 +09:00
59e59ff040 Update README.md atharva-pathak 2025-05-08 10:41:15 +05:30
d1140c1a0f fix pit & add to test Ryan 2025-05-08 02:02:54 +08:00
2e9b2a76fb update some model Ryan 2025-05-08 00:57:18 +08:00
99c25fa5c0 fix nest type error Ryan 2025-05-06 03:43:49 +08:00
72b2a09106 fix final norm only apply at last indice Ryan 2025-05-06 00:56:36 +08:00
e0ae4db8fc update FEAT_INTER_FILTERS list Ryan 2025-05-06 00:25:53 +08:00
98d9cde8b4 support efficientformer_v2 Ryan 2025-05-06 00:25:21 +08:00
12def0d118 support efficientvit, edgenext, davit Ryan 2025-05-06 00:24:57 +08:00
9aedecbb5f fix norm at last feat_idx Ryan 2025-05-05 23:15:39 +08:00
e16d385592 support more models Ryan 2025-05-05 23:11:59 +08:00
6b6beffa6b support mambaout, metaformer, nest, nextvit, pvt_v2 Ryan 2025-05-05 05:30:57 +08:00
2d3155908c fix tresnet and rdnet Ryan 2025-05-05 04:22:11 +08:00
f8be741f0f support rexnet, resnetv2, repvit and repghostnet Ryan 2025-05-05 04:21:12 +08:00
5e8cc616d4 support tiny_vit Ryan 2025-05-05 01:35:02 +08:00
8befebd93c support vovnet Ryan 2025-05-04 23:49:58 +08:00
411b892dbc support tresnet Ryan 2025-05-04 23:49:43 +08:00
fdcf7cf5c3 Check forward_intermediates features against forward_features output Ross Wightman 2025-05-06 12:56:58 -07:00
907a32e699 Check forward_intermediates features against forward_features output check_intermediate_feat Ross Wightman 2025-05-06 12:56:58 -07:00
7fc0692843 update fasternet Ryan 2025-05-06 03:10:19 +08:00
a3e66b14ea add fasternet Ryan 2025-05-06 02:59:33 +08:00
7aeebf20e2 add BEIT3 Ryan 2025-05-06 01:29:55 +08:00
91e6e1737e add test_models Ryan 2025-05-03 23:40:37 +08:00
37bbac1256 Fix checkpoint_filter_fn Ryan 2025-05-03 15:41:18 +08:00
fc0b6ad183 Fix default_cfgs Ryan 2025-05-03 08:40:07 +08:00
848b8c3e57 Support features_only Ryan 2025-05-02 20:59:05 +08:00
b37f0f7a76 Update tnt.py Ryan 2025-05-02 20:34:31 +08:00
e2073e32d0 Move NaFlexCollate with dataset, remove stand alone collate_fn and remove redundancy Ross Wightman 2025-04-29 10:44:46 -07:00
39eb56f875 Starting to test distributed train, fix issue with batch_size reduce Ross Wightman 2025-04-28 16:48:06 -07:00
77788f4f92 add shvit Ryan 2025-04-29 06:24:44 +08:00
b60a4fb329 add starnet Ryan 2025-04-29 05:38:42 +08:00
081e6c2b96 update swiftformer Ryan 2025-04-29 05:19:53 +08:00
7a87199bf8 add swiftformer Ryan 2025-04-29 00:58:10 +08:00
ee27b73da4 Further pos embed tweaks, rejig model defs for testing Ross Wightman 2025-04-28 09:15:11 -07:00
3dc90ed7a7 Add naflex loader support to validate.py, fix bug in naflex pos embed add, classic vit weight loading for naflex model Ross Wightman 2025-04-25 16:00:54 -07:00
c8c4f256b8

add-lightly-train-to-training-frameworks-readme (#2474) Yutong Xiang 2025-04-22 17:50:11 +02:00
c527c37969 Optimizations for pos embed resize, merge different mask helper fns Ross Wightman 2025-04-21 14:05:18 -07:00
0c136f7dab fix img_size type Xihua Dong 2025-04-18 09:13:55 -07:00
3ff3899026 Add local-dir: schema support for model loading (config + weights) from folder Ross Wightman 2025-04-17 10:32:48 -07:00
fe353419af Add local-dir: schema support for model loading (config + weights) from folder local_dir Ross Wightman 2025-04-17 10:32:48 -07:00
ceca5efdec Remove torch_out from onnx export, no point without the export_ fn Ross Wightman 2025-04-15 12:07:19 -07:00
0cae8a4cd8 Fix #2472, torch.onnx.export_ (with return output) finally removed :( Ross Wightman 2025-04-15 12:03:03 -07:00
990f618868 Remove torch_out from onnx export, no point without the export_ fn fix_onnx Ross Wightman 2025-04-15 12:07:19 -07:00
b1d09e2d49 Fix #2472, torch.onnx.export_ (with return output) finally removed :( Ross Wightman 2025-04-15 12:03:03 -07:00
ea728f67fa Improve several typing issues for flex vit, can (almost) work with jit if we bash h,w key into an int or str Ross Wightman 2025-04-14 11:01:56 -07:00
681be882e8 Fix arg merging of sknet, old seresnet. Fix #2470 Ross Wightman 2025-04-14 09:24:31 -07:00
382444362a Fix arg merging of sknet, old seresnet. Fix #2470 sknet_kwargs_merge Ross Wightman 2025-04-14 09:24:31 -07:00
97341fec51 A much faster resample_patch_embed, can be used at train/validation time Ross Wightman 2025-04-10 15:58:24 -07:00
b4bb0f452a Exclude embeds module and mask attn functions from tracing Ross Wightman 2025-04-09 15:34:15 -07:00
13e0f3a4a3 Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model. Ross Wightman 2025-04-08 20:47:57 -07:00
6675590264 Fix ParallelThingsBlock w/ attn_mask Ross Wightman 2025-04-08 09:35:34 -07:00
9b23d6dea2 Exclude naflex models from jit tests Ross Wightman 2025-04-08 07:59:19 -07:00
825edccf19 Type fixes, remove old comments Ross Wightman 2025-04-07 21:35:03 -07:00
0893f5d296 Initial NaFlex ViT model and training support Ross Wightman 2025-04-07 21:27:10 -07:00
e44f14d7d2 Update README v1.0.15 Ross Wightman 2025-02-22 21:04:13 -08:00
98e9651952

Update version.py Ross Wightman 2025-02-22 10:50:21 -08:00
e76ea5474d

Update README.md Ross Wightman 2025-02-21 16:09:42 -08:00
92682d8d4d timm.models: explicitly export attributes Adam J. Stewart 2025-02-21 19:33:00 +01:00