1810 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
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	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
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	98e9651952	Update version.py ... Version 1.0.15, prep for a release	2025-02-22 10:50:21 -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	490d222dd8	Fix issue taking device from V before V exists	2025-01-31 12:52:47 -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
Ross Wightman	872978ccfe	Fix comment, add 'stochastic weight decay' idea because why not	2025-01-30 18:22:36 -08:00
Ross Wightman	510bbd5389	Change start/end args	2025-01-30 18:22:36 -08:00
Ross Wightman	31831f5948	Change flattening behaviour in Kron	2025-01-30 18:22:36 -08:00
Ross Wightman	b3a83b81d6	Prep Kron for merge, add detail to attributions note, README.	2025-01-27 21:02:26 -08:00
Ross Wightman	67ef6f0a92	Move opt_einsum import back out of class __init__	2025-01-27 21:02:26 -08:00
Ross Wightman	9ab5464e4d	More additions to Kron	2025-01-27 21:02:26 -08:00
Ross Wightman	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.	2025-01-27 21:02:26 -08:00
Ross Wightman	cd21e80d03	Fiddling with Kron (PSGD)	2025-01-27 21:02:26 -08:00
Adam J. Stewart	d81da93c16	Use import alias	2025-01-22 10:27:17 -08:00