1372 Commits

Author	SHA1	Message	Date
Ross Wightman	16d0b26e19	Fix torchscript issue with legacy tnt	2025-05-14 09:33:41 -07:00
Ross Wightman	74ad32a67e	Updated tnt model weights on hub, add back legacy model in case bwd compat	2025-05-14 08:40:53 -07:00
Ryan	69b1fbcdc1	Merge branch 'main' into tnt	2025-05-11 22:45:48 +08: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	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
Ryan	37bbac1256	Fix checkpoint_filter_fn	2025-05-03 15:41:18 +08:00
Ryan	fc0b6ad183	Fix default_cfgs	2025-05-03 08:40:07 +08:00
Ryan	848b8c3e57	Support features_only	2025-05-02 20:59:05 +08:00
Ryan	b37f0f7a76	Update tnt.py	2025-05-02 20:34:31 +08: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	681be882e8	Fix arg merging of sknet, old seresnet. Fix #2470	2025-04-14 10:32:26 -07: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	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
Ryan	17eabaad17	Fix RDNet forward call	2025-01-21 11:52:05 -08:00
Ryan	80a4877376	Fix self.reset_classifier num_classes update	2025-01-21 11:52:05 -08:00
Ross Wightman	aa333079da	Tweak so150m2 def	2025-01-19 13:40:53 -08:00
Josua Rieder	8d81fdf3d9	Fix typos	2025-01-19 13:39:40 -08:00
Ross Wightman	3677f67902	Add the 256x256 in1k ft of the so150m, add an alternate so150m def	2025-01-18 15:51:57 -08:00
Ross Wightman	2a84d68d02	Add some so150m vit w/ sbb recipe weights, and a ese_vovnet57b model with RA4 recipe	2025-01-18 15:51:57 -08:00
Ross Wightman	9265d54a3a	LeViT safetensors load is broken by conversion code that wasn't deactivated	2025-01-16 11:37:00 -08:00
Adam J. Stewart	6d21eb0d37	VGG ConvMlp: fix layer defaults/types	2025-01-15 12:11:56 +01:00
Adam J. Stewart	f5c4d5cbb7	Add missing imports	2025-01-11 15:13:16 +01:00
Adam J. Stewart	19aaea3c8f	Fix nn.Module type hints	2025-01-11 15:09:21 +01:00
Ross Wightman	155f6e7fea	Update README, few minor fixups.	2025-01-06 13:09:15 -08:00
Ross Wightman	2b251fb291	Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override	2025-01-06 11:28:39 -08:00
Ross Wightman	e846b2cf28	Add 384x384 in12k pretrain and finetune for convnext_nano	2024-12-31 13:16:43 -08:00
Ross Wightman	b0068ba5d0	Switch hf hub entries for new aimv2 / dfn weights to point to timm locations. Undo forced device for SDR linspace, part of another change.	2024-12-30 19:24:21 -08:00
Ross Wightman	1bf84b35c3	Update tests for aimv2 filtering	2024-12-30 19:24:21 -08:00
Ross Wightman	b33418713a	Add (almost) full set of aimv2 model instances. Switch back to unpacked SwiGLU. Verify correctness. Add DFN L/14 39B weight.	2024-12-30 19:24:21 -08:00
Ross Wightman	5f12a25114	Add bias arg to Vitamin GeGLU	2024-12-30 19:24:21 -08:00
Ross Wightman	5804d92e4b	Switch aimv2 to used packed SwiGLU	2024-12-30 19:24:21 -08:00
Ross Wightman	15406a939e	Fixing RmsNorm to fix #2380 and noticed with aimv2 when comparing outputs. Still some work to do, need to look at AMP / fast mode behaviour, dispatch to torch when possible. Add SimpleNorm for 'LayerNorm w/o centering and bias'	2024-12-30 19:24:21 -08:00
Ross Wightman	a648a04834	Supporting aimv2 encoders	2024-12-30 19:24:21 -08:00