1798 Commits

Author	SHA1	Message	Date
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
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
Adam J. Stewart	4de1abf837	timm: add __all__ to __init__	2025-01-22 10:27:17 -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
Collin McCarthy	84631cb5c6	Add missing training flag to convert_sync_batchnorm	2025-01-21 11:51:55 -08:00
Ross Wightman	5d535d7a2d	Version 1.0.14, update README & changelog	2025-01-19 13:53:09 -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
Ross Wightman	21e75a9d25	Update version.py ... Back to dev version	2025-01-16 11:23:17 -08:00
Adam J. Stewart	6d21eb0d37	VGG ConvMlp: fix layer defaults/types	2025-01-15 12:11:56 +01:00