2738 Commits

Author SHA1 Message Date
Ryan
cb98094792 Merge branch 'main' into fast_model 2025-05-14 08:30:33 +08:00
Ryan
89d2952375 update group_matcher 2025-05-14 08:28:10 +08:00
Ryan Ramos
81900a6bae remove download argument from torch_kwargs 2025-05-09 07:34:32 -07:00
atharva-pathak
59e59ff040 Update README.md
The previous link to Chris Hughes blog post was wrong, so updated the correct link
2025-05-08 08:43:18 -07:00
Ryan
d1140c1a0f fix pit & add to test 2025-05-08 07:17:19 -07:00
Ryan
2e9b2a76fb update some model 2025-05-08 07:17:19 -07:00
Ryan
99c25fa5c0 fix nest type error 2025-05-08 07:17:19 -07:00
Ryan
72b2a09106 fix final norm only apply at last indice 2025-05-08 07:17:19 -07:00
Ryan
e0ae4db8fc update FEAT_INTER_FILTERS list 2025-05-08 07:17:19 -07:00
Ryan
98d9cde8b4 support efficientformer_v2 2025-05-08 07:17:19 -07:00
Ryan
12def0d118 support efficientvit, edgenext, davit 2025-05-08 07:17:19 -07:00
Ryan
9aedecbb5f fix norm at last feat_idx 2025-05-08 07:17:19 -07:00
Ryan
e16d385592 support more models
inception_next, hgnet, gcvit, focalnet, inception_v4
2025-05-08 07:17:19 -07:00
Ryan
6b6beffa6b support mambaout, metaformer, nest, nextvit, pvt_v2 2025-05-08 07:17:19 -07:00
Ryan
2d3155908c fix tresnet and rdnet 2025-05-08 07:17:19 -07:00
Ryan
f8be741f0f support rexnet, resnetv2, repvit and repghostnet 2025-05-08 07:17:19 -07:00
Ryan
5e8cc616d4 support tiny_vit 2025-05-08 07:17:19 -07:00
Ryan
8befebd93c support vovnet 2025-05-08 07:17:19 -07:00
Ryan
411b892dbc support tresnet 2025-05-08 07:17:19 -07:00
Ross Wightman
fdcf7cf5c3 Check forward_intermediates features against forward_features output 2025-05-06 13:48:51 -07:00
Ryan
7fc0692843 update fasternet 2025-05-06 03:10:19 +08:00
Ryan
a3e66b14ea add fasternet 2025-05-06 02:59:33 +08:00
Ryan
91e6e1737e add test_models 2025-05-03 23:40:37 +08:00
Ryan
77788f4f92 add shvit 2025-04-29 06:24:44 +08:00
Ryan
b60a4fb329 add starnet 2025-04-29 05:38:42 +08:00
Ryan
081e6c2b96 update swiftformer 2025-04-29 05:19:53 +08:00
Ryan
7a87199bf8 add swiftformer 2025-04-29 00:58:10 +08:00
Yutong Xiang
c8c4f256b8
add-lightly-train-to-training-frameworks-readme (#2474) 2025-04-22 08:50:11 -07:00
Xihua Dong
0c136f7dab fix img_size type 2025-04-18 21:05:05 -07:00
Ross Wightman
3ff3899026 Add local-dir: schema support for model loading (config + weights) from folder 2025-04-17 12:32:19 -07:00
Ross Wightman
ceca5efdec Remove torch_out from onnx export, no point without the export_ fn 2025-04-15 14:01:17 -07:00
Ross Wightman
0cae8a4cd8 Fix #2472, torch.onnx.export_ (with return output) finally removed :( 2025-04-15 14:01:17 -07:00
Ross Wightman
681be882e8 Fix arg merging of sknet, old seresnet. Fix #2470 2025-04-14 10:32:26 -07:00
Ross Wightman
e44f14d7d2 Update README v1.0.15 2025-02-22 21:04:13 -08:00
Ross Wightman
98e9651952
Update version.py
Version 1.0.15, prep for a release
2025-02-22 10:50:21 -08:00
Ross Wightman
e76ea5474d
Update README.md 2025-02-21 16:09:42 -08:00
Adam J. Stewart
92682d8d4d timm.models: explicitly export attributes 2025-02-21 14:19:39 -08:00
Ross Wightman
a667d3d8f0 siglip2 weights on hub, fix forward_intermediates when no prefix tokens (& return prefix selected) 2025-02-21 13:10:51 -08:00
Ross Wightman
f63a11cf81 Remove duplicate so400m/16 @ 256 model def 2025-02-21 13:10:51 -08:00
Ross Wightman
9758e0b8b0 Prep for siglip2 release 2025-02-21 13:10:51 -08:00
Adam J. Stewart
c68d724e9c adapt_input_conv: add type hints 2025-02-21 12:28:22 -08:00
Ross Wightman
105a667baa Dev version 1.0.15.dev0 2025-02-17 15:50:12 -08:00
Ross Wightman
7234f5c6c5 Add 448 so150m2 weight/model, add updated internvit 300m weight 2025-02-17 12:59:10 -08:00
Ross Wightman
9ce824c39a Add vit so150m2 weights 2025-02-14 15:55:51 -08:00
Ross Wightman
a49b020eff Merge branch 'ClashLuke-patch-1' 2025-01-31 12:53:29 -08:00
Ross Wightman
490d222dd8 Fix issue taking device from V before V exists 2025-01-31 12:52:47 -08:00
Ross Wightman
875c19d0c9 Merge branch 'patch-1' of github.com:ClashLuke/pytorch-image-models into ClashLuke-patch-1 2025-01-31 12:43:28 -08:00
Ross Wightman
8b3c07a841
Update README.md 2025-01-31 10:37:32 -08:00
Lucas Nestler
e025328f96
simplify RNG 2025-01-31 17:26:14 +01:00
Lucas Nestler
6367267298
unify RNG 2025-01-31 17:23:53 +01:00