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 |