mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 06:41:35 +08:00
Code Issues Projects Releases Wiki Activity
2,022 Commits 141 Branches 65 Tags
Commit Graph

9 Commits

Author SHA1 Message Date
Yassine
8ba2038e6b fast_vit: propagate act_layer argument 2023-09-27 10:24:12 -07:00
Ross Wightman
c8b2f28096 Fix a few typos, fix fastvit proj_drop, add code link 2023-08-28 21:26:29 -07:00
Ross Wightman
d6c348765a Fix first_conv for mobileone and fastvit 2023-08-23 22:50:37 -07:00
Ross Wightman
16334e4bec Fix two fastvit issues 2023-08-23 22:50:37 -07:00
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
Ross Wightman
40dbaafef5 Stagify FastViT /w downsample to top of stage 2023-08-23 22:50:37 -07:00
Ross Wightman
8470eb1cb5 More fastvit & mobileone updates, ready for weight upload 2023-08-23 22:50:37 -07:00
Ross Wightman
8474508d07 More work on FastViT, use own impl of MobileOne, validation working with remapped weight, more refactor TODO 2023-08-23 22:50:37 -07:00
Ross Wightman
c7a20cec13 Begin adding FastViT 2023-08-23 22:50:37 -07:00