Commit Graph

Select branches

Hide Pull Requests

0.5.x

0.6.x

adafactor_bv

aimv2

attn_update

augmentation_update

bits_and_tpu

bulk_runner_tweaks

cache_dir

caojiaolong-main

cautious_optim

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

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

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

#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

#2478

#250

#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

Mono Color

• 68174b9367
  Merge 3dc90ed7a7 into c8c4f256b8 Ross Wightman 2025-04-25 23:01:50 +0000
• 3dc90ed7a7 Add naflex loader support to validate.py, fix bug in naflex pos embed add, classic vit weight loading for naflex model naflex Ross Wightman 2025-04-25 16:00:54 -0700
• 61b7922f6b
  Merge b82f9e869c into c8c4f256b8 Bernie Huang 2025-04-25 14:48:46 -0700
• b82f9e869c renaming models and update checkpoint to vit-only berniebear 2025-04-25 21:26:34 +0000
• 3af564f2fb fix default config berniebear 2025-04-25 09:19:18 +0000
• e6bbf9fd56 add default config berniebear 2025-04-25 07:55:00 +0000
• 8eafe2c21e update pe to reuse timm layers berniebear 2025-04-25 06:50:29 +0000
• 0f6e29019c pe integration berniebear 2025-04-24 22:39:32 +0000
• 8e7c4a5bf1
  Merge c9c973b3b5 into c8c4f256b8 Ross Wightman 2025-04-24 21:32:51 +0200
• f393439d2c
  Merge 82f75f4cbb into c8c4f256b8 Bill Psomas 2025-04-23 22:33:03 +0300
• c0a54c4caa
  Merge c3445e9d75 into c8c4f256b8 Thien Tran 2025-04-23 22:31:03 +0300
• ef03c43335
  Merge b283c24f13 into c8c4f256b8 Ryan 2025-04-23 22:42:29 +0800
• 06309819fc
  Merge 6c00fa52b4 into c8c4f256b8 Rorry Brenner 2025-04-22 19:49:59 -0400
• c8c4f256b8
  add-lightly-train-to-training-frameworks-readme (#2474) main Yutong Xiang 2025-04-22 17:50:11 +0200
• c527c37969 Optimizations for pos embed resize, merge different mask helper fns Ross Wightman 2025-04-21 14:05:18 -0700
• 88e8342297
  Merge f63d2e8071 into 0c136f7dab Fredo Guan 2025-04-21 10:47:32 +0200
• 506f12b16e
  Merge 22b1f35dc6 into 0c136f7dab JosuaRieder 2025-04-19 16:48:06 +0000
• 0c136f7dab fix img_size type Xihua Dong 2025-04-18 09:13:55 -0700
• 9035c459ca fix img_size type Xihua Dong 2025-04-18 09:13:55 -0700
• 24dbb34718
  Merge 430f7b4c5d into 3ff3899026 Fredo Guan 2025-04-18 17:13:33 +0200
• 3ff3899026 Add local-dir: schema support for model loading (config + weights) from folder Ross Wightman 2025-04-17 10:32:48 -0700
• fe353419af Add local-dir: schema support for model loading (config + weights) from folder local_dir Ross Wightman 2025-04-17 10:32:48 -0700
• 7685d4f0d1
  Merge 3abb7811fe into ceca5efdec JosuaRieder 2025-04-16 06:38:35 +0800
• ceca5efdec Remove torch_out from onnx export, no point without the export_ fn Ross Wightman 2025-04-15 12:07:19 -0700
• 0cae8a4cd8 Fix #2472, torch.onnx.export_ (with return output) finally removed :( Ross Wightman 2025-04-15 12:03:03 -0700
• 990f618868 Remove torch_out from onnx export, no point without the export_ fn fix_onnx Ross Wightman 2025-04-15 12:07:19 -0700
• b1d09e2d49 Fix #2472, torch.onnx.export_ (with return output) finally removed :( Ross Wightman 2025-04-15 12:03:03 -0700
• 71469258db
  Merge branch 'huggingface:main' into main Yutong Xiang 2025-04-15 15:24:45 +0200
• 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 -0700
• 681be882e8 Fix arg merging of sknet, old seresnet. Fix #2470 Ross Wightman 2025-04-14 09:24:31 -0700
• 382444362a Fix arg merging of sknet, old seresnet. Fix #2470 sknet_kwargs_merge Ross Wightman 2025-04-14 09:24:31 -0700
• 97341fec51 A much faster resample_patch_embed, can be used at train/validation time Ross Wightman 2025-04-10 15:58:24 -0700
• 62d588fcf3
  Merge pull request #1 from lightly-ai/add-lightly-train-to-training-frameworks-readme Yutong Xiang 2025-04-10 16:22:16 +0200
• 37091f0da0 add-lightly-train-to-training-frameworks-readme yutong 2025-04-10 15:32:47 +0200
• b4bb0f452a Exclude embeds module and mask attn functions from tracing Ross Wightman 2025-04-09 15:34:15 -0700
• 13e0f3a4a3 Add loss scale arg, initial distributed loss scale. Maybe fix FX for the model. Ross Wightman 2025-04-08 20:47:57 -0700
• 6675590264 Fix ParallelThingsBlock w/ attn_mask Ross Wightman 2025-04-08 09:35:34 -0700
• 9b23d6dea2 Exclude naflex models from jit tests Ross Wightman 2025-04-08 07:59:19 -0700
• 825edccf19 Type fixes, remove old comments Ross Wightman 2025-04-07 21:35:03 -0700
• 0893f5d296 Initial NaFlex ViT model and training support Ross Wightman 2025-04-07 21:27:10 -0700
• c3445e9d75 support pytorch<2 Thien Tran 2025-03-30 17:16:25 +0800
• d70c481179 override device kwargs of base nn classes Thien Tran 2025-03-30 17:05:25 +0800
• 6c00fa52b4 updated reader_factory to correct extra folders Rorry 2025-03-02 22:09:03 -0500
• cdfcd3d058
  pyproject.toml: add required dep versions Adam J. Stewart 2025-02-23 09:08:06 +0100
• e44f14d7d2 Update README v1.0.15 Ross Wightman 2025-02-22 21:04:13 -0800
• 98e9651952
  Update version.py Ross Wightman 2025-02-22 10:50:21 -0800
• e76ea5474d
  Update README.md Ross Wightman 2025-02-21 16:09:42 -0800
• 92682d8d4d timm.models: explicitly export attributes Adam J. Stewart 2025-02-21 19:33:00 +0100
• a667d3d8f0 siglip2 weights on hub, fix forward_intermediates when no prefix tokens (& return prefix selected) Ross Wightman 2025-02-21 12:46:14 -0800
• f63a11cf81 Remove duplicate so400m/16 @ 256 model def Ross Wightman 2025-02-20 16:14:58 -0800
• 9758e0b8b0 Prep for siglip2 release Ross Wightman 2025-02-20 12:07:49 -0800
• 228e080e39 siglip2 weights on hub, fix forward_intermediates when no prefix tokens (& return prefix selected) siglip2 Ross Wightman 2025-02-21 12:46:14 -0800
• c68d724e9c adapt_input_conv: add type hints Adam J. Stewart 2025-02-21 19:24:05 +0100
• b1ee8fb96c
  timm.models: explicitly export attributes Adam J. Stewart 2025-02-21 19:33:00 +0100
• d2cbc7fb21
  adapt_input_conv: add type hints Adam J. Stewart 2025-02-21 19:24:05 +0100
• 25de6b4ee1 Remove duplicate so400m/16 @ 256 model def Ross Wightman 2025-02-20 16:14:58 -0800
• dc5c7989ae Prep for siglip2 release Ross Wightman 2025-02-20 12:07:49 -0800
• 105a667baa Dev version 1.0.15.dev0 Ross Wightman 2025-02-17 15:50:12 -0800
• 7234f5c6c5 Add 448 so150m2 weight/model, add updated internvit 300m weight Ross Wightman 2025-02-17 12:59:10 -0800
• 9ce824c39a Add vit so150m2 weights Ross Wightman 2025-02-14 15:02:32 -0800
• 5942f1d492 Add vit so150m2 weights so150m2 Ross Wightman 2025-02-14 15:02:32 -0800
• a49b020eff Merge branch 'ClashLuke-patch-1' Ross Wightman 2025-01-31 12:53:29 -0800
• 490d222dd8 Fix issue taking device from V before V exists Ross Wightman 2025-01-31 12:52:27 -0800
• 875c19d0c9 Merge branch 'patch-1' of github.com:ClashLuke/pytorch-image-models into ClashLuke-patch-1 Ross Wightman 2025-01-31 12:43:28 -0800
• 8b3c07a841
  Update README.md Ross Wightman 2025-01-31 10:37:32 -0800
• e025328f96
  simplify RNG Lucas Nestler 2025-01-31 17:26:14 +0100
• 6367267298
  unify RNG Lucas Nestler 2025-01-31 17:23:53 +0100
• 872978ccfe Fix comment, add 'stochastic weight decay' idea because why not Ross Wightman 2025-01-30 15:42:27 -0800
• 510bbd5389 Change start/end args Ross Wightman 2025-01-30 13:13:49 -0800
• 31831f5948 Change flattening behaviour in Kron Ross Wightman 2025-01-30 13:07:20 -0800
• 5f85f8eefa Fix comment, add 'stochastic weight decay' idea because why not kron_flatten Ross Wightman 2025-01-30 15:42:27 -0800
• 5940cc167f Change start/end args Ross Wightman 2025-01-30 13:13:49 -0800
• 3be8b1abe4 Change flattening behaviour in Kron Ross Wightman 2025-01-30 13:07:20 -0800
• cdbafd9057 Try to force numpy<2.0 for torch 1.13 tests, update newest tested torch to 2.5.1 Ross Wightman 2025-01-28 14:51:25 -0800
• 93f44d1805 Try to force numpy<2.0 for torch 1.13 tests, update newest tested torch to 2.5.1 update_test_workflow Ross Wightman 2025-01-28 14:51:25 -0800
• b1752eefb5 Fix missing model key in bulk validate results on error Ross Wightman 2025-01-28 13:20:40 -0800
• b3a83b81d6 Prep Kron for merge, add detail to attributions note, README. Ross Wightman 2025-01-27 16:00:58 -0800
• 67ef6f0a92 Move opt_einsum import back out of class __init__ Ross Wightman 2025-01-27 14:03:25 -0800
• 9ab5464e4d More additions to Kron Ross Wightman 2025-01-27 13:09:09 -0800
• 5f10450235 Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now. Ross Wightman 2025-01-26 20:44:09 -0800
• cd21e80d03 Fiddling with Kron (PSGD) Ross Wightman 2025-01-24 14:16:21 -0800
• 9c26c959eb Prep Kron for merge, add detail to attributions note, README. kron_optimizer Ross Wightman 2025-01-27 16:00:58 -0800
• 80a0205725 Move opt_einsum import back out of class __init__ Ross Wightman 2025-01-27 14:03:25 -0800
• 71d174180f More additions to Kron Ross Wightman 2025-01-27 13:09:09 -0800
• f759d1213a Some more kron work. Figured out why some tests fail, implemented a deterministic rng state load but too slow so skipping some tests for now. Ross Wightman 2025-01-26 20:44:09 -0800
• de2f5c6a05 Fiddling with Kron (PSGD) Ross Wightman 2025-01-24 14:16:21 -0800
• b283c24f13 Update T2T_ViT Ryan 2025-01-24 00:05:22 +0800
• bd46b550de Add T2T_ViT Ryan 2025-01-23 04:31:22 +0800
• d81da93c16 Use import alias Adam J. Stewart 2025-01-22 10:08:18 +0100
• 4de1abf837 timm: add __all__ to __init__ Adam J. Stewart 2025-01-11 14:31:41 +0100
• 1e189be46d
  Use import alias Adam J. Stewart 2025-01-22 10:08:18 +0100
• bda46f8e6f Add num_classes assertion after reset_classifier Ryan 2025-01-22 01:43:32 +0800
• 17eabaad17 Fix RDNet forward call Ryan 2025-01-22 00:41:04 +0800
• 80a4877376 Fix self.reset_classifier num_classes update Ryan 2025-01-22 00:40:16 +0800
• 84631cb5c6 Add missing training flag to convert_sync_batchnorm Collin McCarthy 2025-01-21 11:41:48 -0800
• 611775581a Add missing training flag to convert_sync_batchnorm Collin McCarthy 2025-01-21 11:41:48 -0800
• a31bb8a879 Add num_classes assertion after reset_classifier Ryan 2025-01-22 01:43:32 +0800
• 61bf9e1af1 Fix RDNet forward call Ryan 2025-01-22 00:41:04 +0800
• c0b12fcb16 Fix self.reset_classifier num_classes update Ryan 2025-01-22 00:40:16 +0800
• a8b90bf5e9 monkeypatch ml-decoder + metaformer Fredo Guan 2025-01-21 01:38:35 -0700