12 Commits

Author	SHA1	Message	Date
Ross Wightman	5242ba6edc	MobileOne and FastViT weights on HF hub, more code cleanup and tweaks, features_only working. Add reparam flag to validate and benchmark, support reparm of all models with fuse(), reparameterize() or switch_to_deploy() methods on modules	2023-08-23 22:50:37 -07:00
Fredo Guan	81ca323751	Davit update formatting and fix grad checkpointing (#7) ... fixed head to gap->norm->fc as per convnext, along with option for norm->gap->fc failed tests due to clip convnext models, davit tests passed	2023-01-15 14:34:56 -08:00
Ross Wightman	57992509f9	Fix some formatting in utils/model.py	2021-10-23 20:35:36 -07:00
Ross Wightman	e5da481073	Small post-merge tweak for freeze/unfreeze, add to __init__ for utils	2021-10-06 17:00:27 -07:00
Alexander Soare	431e60c83f	Add acknowledgements for freeze_batch_norm inspiration	2021-10-06 14:28:49 +01:00
Alexander Soare	65c3d78b96	Freeze unfreeze functionality finalized. Tests added	2021-10-02 15:55:08 +01:00
Alexander Soare	0cb8ea432c	wip	2021-10-02 15:55:08 +01:00
Aman Arora	6b18061773	Add GIST to docstring for quick access	2021-03-29 15:33:31 +11:00
Aman Arora	92b1db9a79	update docstrings and add check on and	2021-03-29 10:04:51 +11:00
Aman Arora	b85be24054	update to work with fnmatch	2021-03-29 09:36:31 +11:00
Aman Arora	20626e8387	Add to extract stats for SPP	2021-03-27 05:40:04 +11:00
Ross Wightman	532e3b417d	Reorg of utils into separate modules	2020-09-07 13:58:09 -07:00