huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-21 12:20:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

FastViT fixes. (#2452)

Browse Source 
* 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
Download Patch File Download Diff File Expand all files Collapse all files

10
candle-examples/examples/fastvit/README.md
View File

@ -12,9 +12,9 @@ $ cargo run --example fastvit --release -- --image candle-examples/examples/yolo
loaded image Tensor[dims 3, 256, 256; f32]
model built
mountain bike, all-terrain bike, off-roader: 43.45%
bicycle-built-for-two, tandem bicycle, tandem: 14.16%
unicycle, monocycle : 4.12%
crash helmet : 2.26%
alp : 1.40%
mountain bike, all-terrain bike, off-roader: 52.67%
bicycle-built-for-two, tandem bicycle, tandem: 7.93%
unicycle, monocycle : 3.46%
maillot : 1.32%
crash helmet : 1.28%
```