mirrors
/
pytorch-image-models
mirror of https://github.com/huggingface/pytorch-image-models.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
1,859 Commits
136 Branches
65 Tags
47 MiB
Commit Graph

1859 Commits (8c6fccb879954a3715a73816f762aa3c36d2a636)
 

Author SHA1 Message Date
Ross Wightman 086bd55a94 Add EfficientFormer-V2, refactor EfficientFormer and Levit for more uniformity across the 3 related arch. Add features_out support to levit conv models and efficientformer_v2. All weights on hub. 2023-02-03 14:12:29 -08:00
Ross Wightman 2cb2699dc8 Apply fix from #1649 to main 2023-02-03 11:28:57 -08:00
Ross Wightman e0a5911072
Merge pull request #1645 from rwightman/norm_mlp_classifier 
Extract NormMlpClassifierHead from maxxvit.py
 2023-02-03 11:00:58 -08:00
Ross Wightman b3042081b4 Add laion -> in1k fine-tuned base and large_mlp weights for convnext 2023-02-03 10:58:02 -08:00
Ross Wightman 316bdf8955 Add mlp head support for convnext_large, add laion2b CLIP weights, prep fine-tuned weight tags 2023-02-01 08:27:02 -08:00
Ross Wightman 6f28b562c6 Factor NormMlpClassifierHead from MaxxViT and use across MaxxViT / ConvNeXt / DaViT, refactor some type hints & comments 2023-01-27 14:57:01 -08:00
Ross Wightman 29fda20e6d Merge branch 'fffffgggg54-main' 2023-01-27 13:55:17 -08:00
Ross Wightman 9a53c3f727 Finalize DaViT, some formatting and modelling simplifications (separate PatchEmbed to Stem + Downsample, weights on HF hub. 2023-01-27 13:54:04 -08:00
Fredo Guan fb717056da Merge remote-tracking branch 'upstream/main' 2023-01-26 10:49:15 -08:00
Ross Wightman 2bbc26dd82 version 0.8.8dev0 2023-01-25 18:02:48 -08:00
Ross Wightman 64667bfa0e Add 'gigantic' vit clip variant for feature extraction and future fine-tuning 2023-01-25 18:02:10 -08:00
Ross Wightman 3aa31f537d
Merge pull request #1641 from rwightman/maxxvit_hub 
MaxxViT weights on hub, new 12k FT 1k weights, convnext 384x384 12k FT 1k, and more
 2023-01-20 20:26:54 -08:00
Ross Wightman 9983ed7721 xlarge maxvit killing the tests 2023-01-20 16:16:20 -08:00
Ross Wightman c2822568ec Update version to 0.8.7dev0 2023-01-20 15:01:10 -08:00
Ross Wightman 0417a9dd81 Update README 2023-01-20 15:00:49 -08:00
Ross Wightman 36989cfae4 Factor out readme generation in hub helper, add more readme fields 2023-01-20 14:49:40 -08:00
Ross Wightman 32f252381d Change order of checkpoitn filtering fn application in builder, try dict, model variant first 2023-01-20 14:48:54 -08:00
Ross Wightman e9f1376cde Cleanup resolve data config fns, add 'model' variant that takes model as first arg, make 'args' arg optional in original fn 2023-01-20 14:47:55 -08:00
Ross Wightman bed350f5e5 Push all MaxxViT weights to HF hub, cleanup impl, add feature map extraction support and prompote to 'std' architecture. Fix norm head for proper embedding / feat map output. Add new in12k + ft 1k weights. 2023-01-20 14:45:25 -08:00
Ross Wightman ca38e1e73f Update ClassifierHead module, add reset() method, update in_chs -> in_features for consistency 2023-01-20 14:44:05 -08:00
Ross Wightman 8ab573cd26 Add convnext_tiny and convnext_small 384x384 fine-tunes of in12k weights, fix pool size for laion CLIP convnext weights 2023-01-20 14:40:16 -08:00
Fredo Guan e58a884c1c Merge remote-tracking branch 'upstream/main' 2023-01-15 14:37:52 -08: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 e9aac412de Correct mean/std for CLIP convnexts 2023-01-14 22:53:56 -08:00
Ross Wightman 42bd8f7bcb Add convnext_base CLIP image tower weights for fine-tuning / features 2023-01-14 21:16:29 -08:00
Ross Wightman 65aea97067
Update tests.yml 
Attempt to work around flaky azure ubuntu mirrors
 2023-01-12 21:31:44 -08:00
Ross Wightman dd60c45044
Merge pull request #1633 from rwightman/freeze_norm_revisit 
Update batchnorm freezing to handle NormAct variants
 2023-01-12 21:13:58 -08:00
Ross Wightman e520553e3e Update batchnorm freezing to handle NormAct variants, Add GroupNorm1Act, update BatchNormAct2d tracing change from PyTorch 2023-01-12 16:55:47 -08:00
Ross Wightman a2c14c2064 Add tiny/small in12k pretrained and fine-tuned ConvNeXt models 2023-01-11 14:50:39 -08:00
Ross Wightman c061d5e401 Allow using class_map functionality w/ IterableDataset (TFDS/WDS) to remap class labels 2023-01-09 16:28:47 -08:00
Ross Wightman 01fdf44438 Initial focalnet import, more refactoring needed for timm. 2023-01-09 16:18:19 -08:00
Ross Wightman 01aea8c1bf Version 0.8.6dev0 2023-01-09 13:38:31 -08:00
Ross Wightman 2e83bba142 Revert head norm changes to ConvNeXt as it broke some downstream use, alternate workaround for fcmae weights 2023-01-09 13:37:40 -08:00
Ikko Eltociear Ashimine 2c24cb98f1 Fix typo in results/README.md 
occuring -> occurring
 2023-01-09 12:23:39 -08:00
Ross Wightman 1825b5e314 maxxvit type 2023-01-09 08:57:31 -08:00
Ross Wightman 5078b28f8a More kwarg handling tweaks, maxvit_base_rw def added 2023-01-09 08:57:31 -08:00
Ross Wightman c0d7388a1b Improving kwarg merging in more models 2023-01-09 08:57:31 -08:00
Ross Wightman 94a91598c3
Update README.md 2023-01-06 21:39:25 -08:00
Ross Wightman d2ef5a3a94
Update README.md 2023-01-06 21:38:40 -08:00
Ross Wightman ae9153052f
Update version.py 2023-01-06 17:17:35 -08:00
Ross Wightman 60ebb6cefa Re-order vit pretrained entries for more sensible default weights (no .tag specified) 2023-01-06 16:12:33 -08:00
Ross Wightman e861b74cf8 Pass through --model-kwargs (and --opt-kwargs for train) from command line through to model __init__. Update some models to improve arg overlay. Cleanup along the way. 2023-01-06 16:12:33 -08:00
Ross Wightman add3fb864e Working on improved model card template for push_to_hf_hub 2023-01-06 16:12:33 -08:00
Xa9aX ツ 13c7183c52 Update installation.mdx 2023-01-06 09:16:43 -08:00
Ross Wightman eb83eb3bd1 Rotate changelogs, add redirects to mkdocs -> equivalent HF docs pages 2023-01-05 17:27:13 -08:00
Ross Wightman dd0bb327e9
Update version.py 
Ver 0.8.4dev0
 2023-01-05 07:55:18 -08:00
Ross Wightman 6e5553da5f
Add ConvNeXt-V2 support (model additions and weights) (#1614) 
* Add ConvNeXt-V2 support (model additions and weights)

* ConvNeXt-V2 weights on HF Hub, tweaking some tests

* Update README, fixing convnextv2 tests
 2023-01-05 07:53:32 -08:00
nateraw 3698e79ac5 🐛 fix github source links in hf docs 2023-01-04 09:42:05 -08:00
Nathan Raw 9f5bba9ef9
Structure Hugging Face Docs (#1575) 
* 🎨 structure docs

* 🚧 wip docs

* 📝 add installation doc

* 📝 wip docs

* 📝 wip docs

* 📝 wip docs

* 📝 wip docs

* 📝 wip docs

* 📝 add basic reference docs

* 📝 remove augmentation from toctree

* 👷 update pr doc builder to bugfix branch

* 📝 wip docs

* 🚧 wip

* 👷 bump CI

* 🚧 wip

* 🚧 bump CI

* 🚧 wip

* 🚧 wip

* 🚧 wip

* 📝 add hf hub tutorial doc

* 🔥 remove inference tut

* 🚧 wip

* 📝 wip docs

* 📝 wip docs

* 📝 update docs

* 📝 move validation script doc up in order

* 🎨 restructure to remove legacy docs

* 📝 update index doc

* 📝 update number of pretrained models

* Update hfdocs/README.md

* Update .github/workflows/build_pr_documentation.yml

* Update build_pr_documentation.yml

* bump

* 📌 update gh action to use main branch

* 🔥 remove comment
 2023-01-03 14:13:53 -08:00
Ross Wightman 960f5f92e6 Update results csv with latest val/test set runs 2022-12-30 15:42:41 -08:00