pytorch-image-models

Commit Graph

Select branches

Hide Pull Requests

0.5.x

0.6.x

attn_update

bits_and_tpu

cleanup_xla_model_fixes

clip_laion2b

convnext_and_copyright

convnextv2

dataset_info

edgenext_csp_and_more

efficientnet_attn

eva

fix_tests

fix_window_trace

fixes-syncbn_pretrain_cfg_resolve

fixes_bce_regnet

focalnet_and_swin_refactor

freeze_norm_revisit

gh-pages

hf_hub_revisit

levit_efficientformer_redux

main

maxxvit_hub

more_attn

more_datasets

more_vit

multi-weight

multi-weight_effnet_convnext

norm_mlp_classifier

norm_norm_norm

parse_kwargs

parse_kwargs2

perceiver

refactor-imports

scaling_vit

script_cleanup

swin_v2

update-test-pytorch

vit_and_bit_test_fixes

vit_relpos_refactor

xsepconv

#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

#1210

#1212

#1213

#1213

#122

#1222

#1222

#1229

#123

#1230

#1233

#1239

#1245

#1249

#125

#1251

#1253

#1253

#1256

#1259

#1265

#1266

#1270

#1271

#1271

#1273

#1274

#1276

#1277

#1294

#13

#1317

#132

#1322

#1327

#1330

#1330

#1336

#1340

#1345

#1349

#1351

#1354

#136

#1363

#1365

#137

#1381

#1383

#1385

#140

#141

#1414

#1415

#142

#1420

#143

#1439

#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

#1553

#156

#1562

#1564

#1565

#1575

#1578

#1578

#1581

#1582

#1583

#1586

#1590

#1590

#1592

#1593

#16

#1606

#1612

#1614

#1617

#1617

#1621

#1622

#1624

#1627

#1628

#1628

#163

#1630

#1633

#1641

#1643

#1643

#1645

#1647

#1647

#1649

#1654

#1654

#1655

#1659

#1659

#1662

#1673

#1674

#1677

#1679

#1679

#1680

#1683

#1685

#1687

#1688

#1688

#175

#179

#183

#19

#191

#195

#199

#2

#208

#212

#212

#213

#213

#214

#216

#218

#227

#228

#23

#233

#237

#24

#240

#240

#244

#250

#255

#256

#263

#268

#282

#284

#286

#29

#290

#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

#603

#609

#612

#613

#62

#625

#626

#636

#637

#645

#647

#65

#651

#652

#653

#659

#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

#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.5

v0.6.7

v0.8.10dev0

v0.8.13dev0

v0.8.2dev0

v0.8.6dev0

e9f1376cde Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn Ross Wightman 2023-01-20 14:47:55 -0800
bed350f5e5 Push all MaxxViT weights to HF hub, cleanup impl, add feature map extraction support and prompote to 'std' architecture. Fix norm head for proper embedding / feat map output. Add new in12k + ft 1k weights. Ross Wightman 2023-01-20 14:45:25 -0800
ca38e1e73f Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency Ross Wightman 2023-01-20 14:44:05 -0800
8ab573cd26 Add convnext_tiny and convnext_small 384x384 fine-tunes of in12k weights, fix pool size for laion CLIP convnext weights Ross Wightman 2023-01-20 14:40:16 -0800
729533e966 Update metaformers.py Fredo Guan 2023-01-20 02:01:06 -0800
dd57cde1bc Update metaformers.py Fredo Guan 2023-01-20 01:56:03 -0800
143f8e69b1 Update metaformers.py Fredo Guan 2023-01-20 01:53:36 -0800
61b4e716ca Update metaformers.py Fredo Guan 2023-01-18 12:30:37 -0800
9a415a6041 Update metaformers.py Fredo Guan 2023-01-18 12:29:42 -0800
53f992723c Update metaformers.py Fredo Guan 2023-01-18 12:25:07 -0800
383c9fd43d Update metaformers.py Fredo Guan 2023-01-18 12:12:26 -0800
eecab9eba8 Update metaformers.py Fredo Guan 2023-01-18 12:05:16 -0800
9efb706b25 Update metaformers.py Fredo Guan 2023-01-18 12:03:11 -0800
3c6821cc12 Update metaformers.py Fredo Guan 2023-01-18 11:55:16 -0800
3685404604 Update metaformers.py Fredo Guan 2023-01-18 11:52:18 -0800
a51d1073f5 Update metaformers.py Fredo Guan 2023-01-18 11:21:18 -0800
3de6859d9b Update metaformers.py Fredo Guan 2023-01-18 11:17:15 -0800
118017ce3d Update metaformers.py Fredo Guan 2023-01-18 11:10:02 -0800
debf4e9171 Update metaformers.py Fredo Guan 2023-01-18 11:07:40 -0800
f2c4d6f963 Update metaformers.py Fredo Guan 2023-01-18 11:04:30 -0800
6202d9c898 Update metaformers.py Fredo Guan 2023-01-18 10:49:32 -0800
93ed33bc43 Update metaformers.py Fredo Guan 2023-01-18 10:41:22 -0800
924a64f051 Update metaformers.py Fredo Guan 2023-01-18 09:30:16 -0800
0cf0cc837c Update metaformers.py Fredo Guan 2023-01-18 08:25:10 -0800
0c89c7eb2d rename model Fredo Guan 2023-01-18 00:16:50 -0800
cd36989a60 Update metaformers.py Fredo Guan 2023-01-17 20:57:51 -0800
7babbfa7a4 Merge branch 'metaformer_baselines_for_vision' of https://github.com/fffffgggg54/pytorch-image-models into metaformer_baselines_for_vision Fredo Guan 2023-01-17 13:23:02 -0800
49bf08ed22 Update metaformers.py Fredo Guan 2023-01-17 13:22:16 -0800
47b718a1c8 Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:34:29 -0800
869e01cec6 Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:34:01 -0800
f985f35b91 Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:33:56 -0800
a1882204ea Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:33:51 -0800
2912d8c477 Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:33:44 -0800
4f6bbbfb0e Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:33:36 -0800
e4f89b2e25 Revert "Update metaformers.py" Fredo Guan 2023-01-17 12:33:31 -0800
2916f37f8d Update metaformers.py Fredo Guan 2023-01-17 12:27:12 -0800
1d882eb494 Update metaformers.py Fredo Guan 2023-01-17 12:22:29 -0800
2209d0830e Update metaformers.py Fredo Guan 2023-01-17 12:20:54 -0800
32bede4e27 Update metaformers.py Fredo Guan 2023-01-17 12:18:20 -0800
4ed934e000 Update metaformers.py Fredo Guan 2023-01-17 12:14:45 -0800
3f0b075367 Update metaformers.py Fredo Guan 2023-01-17 12:00:17 -0800
cf28c57fda Update metaformers.py Fredo Guan 2023-01-17 11:58:27 -0800
2fef9006d7 Update metaformers.py Fredo Guan 2023-01-17 11:56:52 -0800
6b6510f30a Update metaformers.py Fredo Guan 2023-01-17 11:49:10 -0800
808f4a7ebd Update metaformers.py Fredo Guan 2023-01-17 11:13:14 -0800
bee9576f03 Update metaformers.py Fredo Guan 2023-01-17 11:11:19 -0800
9bf9d163bc Update metaformers.py Fredo Guan 2023-01-17 11:08:25 -0800
13876ada4c Update metaformers.py Fredo Guan 2023-01-17 11:05:30 -0800
473403d905 Update metaformers.py Fredo Guan 2023-01-17 11:00:25 -0800
09d1ea628d Update metaformers.py Fredo Guan 2023-01-17 10:58:07 -0800
bfa6f0962d Update metaformers.py Fredo Guan 2023-01-17 10:56:03 -0800
5fe33b33a0 Merge branch 'metaformer_baselines_for_vision' of https://github.com/fffffgggg54/pytorch-image-models into metaformer_baselines_for_vision Fredo Guan 2023-01-17 10:49:36 -0800
5de2544a80 Update metaformers.py Fredo Guan 2023-01-17 10:49:34 -0800
eaf54b66af Update metaformers.py Fredo Guan 2023-01-17 08:01:43 -0800
55d4eb78a2 merge with poolformer, initial version Fredo Guan 2023-01-17 07:47:52 -0800
e58a884c1c Merge remote-tracking branch 'upstream/main' Fredo Guan 2023-01-15 14:37:52 -0800
81ca323751

Davit update formatting and fix grad checkpointing (#7) Fredo Guan 2023-01-15 14:34:56 -0800
e9aac412de Correct mean/std for CLIP convnexts Ross Wightman 2023-01-14 22:53:56 -0800
42bd8f7bcb Add convnext_base CLIP image tower weights for fine-tuning / features Ross Wightman 2023-01-14 21:16:29 -0800
16664ea31b

Merge branch 'rwightman:main' into master #1330 swenkel 2023-01-13 09:37:10 +0200
eab434babe

Merge 0f75d48a66 into 65aea97067 #212 Santiago Castro 2023-01-13 15:48:25 +0900
65aea97067

Update tests.yml Ross Wightman 2023-01-12 21:31:44 -0800
dd60c45044

Merge pull request #1633 from rwightman/freeze_norm_revisit Ross Wightman 2023-01-12 21:13:58 -0800
e520553e3e Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch #1633 freeze_norm_revisit Ross Wightman 2023-01-12 16:55:47 -0800
5f2bebd7ca Update metaformers.py Fredo Guan 2023-01-12 11:37:33 -0800
2a9f93c064 Update metaformers.py Fredo Guan 2023-01-12 11:36:18 -0800
087a451337 Update metaformers.py Fredo Guan 2023-01-12 11:32:10 -0800
1b1b1d83b4 Update metaformers.py Fredo Guan 2023-01-12 11:24:19 -0800
f400e8a3c9 Update metaformers.py Fredo Guan 2023-01-12 11:21:43 -0800
d40d4c8c2a Update metaformers.py Fredo Guan 2023-01-12 11:16:05 -0800
7869dd6769 Update metaformers.py Fredo Guan 2023-01-12 11:13:17 -0800
a73e464bc8 Update metaformers.py Fredo Guan 2023-01-12 11:12:05 -0800
0a9c2607a0 Update metaformers.py Fredo Guan 2023-01-12 11:10:48 -0800
199b443884 Update metaformers.py Fredo Guan 2023-01-12 11:08:18 -0800
61e8414ad0 Update metaformers.py Fredo Guan 2023-01-12 11:02:43 -0800
95ec7cf016 Update metaformers.py Fredo Guan 2023-01-12 11:01:28 -0800
a776d98d3f Update metaformers.py Fredo Guan 2023-01-12 10:52:50 -0800
38168b0741 Update davit.py #1630 Fredo Guan 2023-01-11 20:59:14 -0800
a2c14c2064 Add tiny/small in12k pretrained and fine-tuned ConvNeXt models v0.8.6dev0 Ross Wightman 2023-01-11 14:50:39 -0800
c061d5e401 Allow using class_map functionality w/ IterableDataset (TFDS/WDS) to remap class labels Ross Wightman 2023-01-09 16:28:47 -0800
01fdf44438 Initial focalnet import, more refactoring needed for timm. Ross Wightman 2023-01-09 16:18:19 -0800
01aea8c1bf Version 0.8.6dev0 Ross Wightman 2023-01-09 13:38:31 -0800
2e83bba142 Revert head norm changes to ConvNeXt as it broke some downstream use, alternate workaround for fcmae weights Ross Wightman 2023-01-09 13:37:40 -0800
2c24cb98f1 Fix typo in results/README.md Ikko Eltociear Ashimine 2023-01-10 01:26:57 +0900
1825b5e314 maxxvit type Ross Wightman 2023-01-08 22:42:24 -0800
5078b28f8a More kwarg handling tweaks, maxvit_base_rw def added Ross Wightman 2023-01-08 18:17:17 -0800
c0d7388a1b Improving kwarg merging in more models Ross Wightman 2023-01-07 16:29:12 -0800
4bca10d0b5

Fix typo in results/README.md #1627 Ikko Eltociear Ashimine 2023-01-10 01:26:57 +0900
7f5393f94d maxxvit type #1624 parse_kwargs2 Ross Wightman 2023-01-08 22:42:24 -0800
8968a03ed4 More kwarg handling tweaks, maxvit_base_rw def added Ross Wightman 2023-01-08 18:17:17 -0800
8dbba278b7 Update metaformers.py Fredo Guan 2023-01-08 12:37:01 -0800
3e3010b65e Update metaformers.py Fredo Guan 2023-01-08 12:35:08 -0800
cc8575e9d1 Update metaformers.py Fredo Guan 2023-01-08 12:33:57 -0800
bde9554604 Update metaformers.py Fredo Guan 2023-01-08 12:30:10 -0800
464ae0fe50 Update metaformers.py Fredo Guan 2023-01-08 12:27:57 -0800
8568bc7b6a Update metaformers.py Fredo Guan 2023-01-08 12:13:14 -0800
ddfdb543dc Update metaformers.py Fredo Guan 2023-01-08 11:53:20 -0800
4e682f1fda Update metaformers.py Fredo Guan 2023-01-08 11:52:13 -0800
bc37dec29b Update metaformers.py Fredo Guan 2023-01-08 11:51:19 -0800
0aa784027e Update metaformers.py Fredo Guan 2023-01-08 11:22:55 -0800