Ross Wightman
|
5f81d4de23
|
Move DeiT to own file, vit getting crowded. Working towards fixing #1029, make pooling interface for transformers and mlp closer to convnets. Still working through some details...
|
2022-01-26 22:53:57 -08:00 |
|
Ross Wightman
|
abc9ba2544
|
Transitioning default_cfg -> pretrained_cfg. Improving handling of pretrained_cfg source (HF-Hub, files, timm config, etc). Checkpoint handling tweaks.
|
2022-01-25 21:54:13 -08:00 |
|
Ross Wightman
|
f2006b2437
|
Cleanup qkv_bias cat in beit model so it can be traced
|
2021-11-18 21:25:00 -08:00 |
|
Thomas Viehmann
|
f805ba86d9
|
use .unbind instead of explicitly listing the indices
|
2021-10-24 21:08:47 +02:00 |
|
Ross Wightman
|
1c9284c640
|
Add BeiT 'finetuned' 1k weights and pretrained 22k weights, pretraining specific (masked) model excluded for now
|
2021-09-13 16:38:23 -07:00 |
|