FastViT fixes. (#2452)

* correct optional SE layer dimensions.
 * head_dim instead of num_heads is 32.
 * update test example output.
This commit is contained in:
Jani Monoses
2024-08-28 12:20:09 +03:00
committed by GitHub
parent aafa24ed93
commit 29e25c458d
3 changed files with 9 additions and 9 deletions

View File

@ -241,7 +241,7 @@ If you have an addition to this list, please submit a pull request.
- Parler-TTS, text-to-speech model.
- Computer Vision Models.
- DINOv2, ConvMixer, EfficientNet, ResNet, ViT, VGG, RepVGG, ConvNeXT,
ConvNeXTv2, MobileOne, EfficientVit (MSRA), MobileNetv4, Hiera.
ConvNeXTv2, MobileOne, EfficientVit (MSRA), MobileNetv4, Hiera, FastViT.
- yolo-v3, yolo-v8.
- Segment-Anything Model (SAM).
- SegFormer.