Ross Wightman
|
9709dbaaa9
|
Adding support for fine-tune CLIP LAION-2B image tower weights for B/32, L/14, H/14 and g/14. Still WIP
|
2 years ago |
Ross Wightman
|
4f0f9cb348
|
Fix #954 by bringing traceable _assert into timm to allow compat w/ PyTorch < 1.8
|
3 years ago |
Ross Wightman
|
b745d30a3e
|
Fix formatting of last commit
|
3 years ago |
Ross Wightman
|
3478f1d7f1
|
Traceability fix for vit models for some experiments
|
3 years ago |
Ross Wightman
|
ecc7552c5c
|
Add levit, levit_c, and visformer model defs. Largely untested and not finished cleanup.
|
4 years ago |
Ross Wightman
|
715519a5ef
|
Rethink name of patch embed grid info
|
4 years ago |
Ross Wightman
|
b2c305c2aa
|
Move Mlp and PatchEmbed modules into layers. Being used in lots of models now...
|
4 years ago |