c838c4233f
Add typing to reset_classifier() on other models
2024-05-12 11:12:00 -07:00
d6c2cc91af
Make NormMlpClassifier head reset args consistent with ClassifierHead
2024-02-10 16:25:33 -08:00
95ba90157f
Update tiny_vit.py to fix bug
2023-09-23 10:05:52 -07:00
0d124ffd4f
Update README. Fine-grained layer-wise lr decay working for tiny_vit and both efficientvits. Minor fixes.
2023-09-01 15:05:29 -07:00
2f0fbb59b3
TinyViT weights on HF hub
2023-09-01 11:05:56 -07:00
507cb08acf
TinyVitBlock needs adding as leaf for FX now, tweak a few dim names
2023-09-01 11:05:56 -07:00
9caf32b93f
Move levit style pos bias resize with other rel pos bias utils
2023-09-01 11:05:56 -07:00
63417b438f
TinyViT adjustments
...
* keep most of net in BCHW layout, performance appears same, can remove static resolution attribs and features easier to use
* add F.sdpa, decent gains in pt 2.1
* tweak crop pct based on eval
2023-09-01 11:05:56 -07:00
39aa44b192
Fixing tinyvit trace issue
2023-09-01 11:05:56 -07:00
aea3b9c854
Fixing tinyvit input_size issue
2023-09-01 11:05:56 -07:00
fabc4e5bcd
Fixing tinyvit torchscript issue
2023-09-01 11:05:56 -07:00
bae949f830
fix attention_bias_cache in tinyvit
2023-09-01 11:05:56 -07:00
170a5b6e27
add tinyvit
2023-09-01 11:05:56 -07:00
Ross Wightman
Nguyen Nhat Hoang
方曦