Phil Wang e8ca6038c9 allow for DistillableVit to still run predictions

2021-01-11 10:49:14 -08:00

..

__init__.py

first commit

2020-10-04 12:35:01 -07:00

distill.py

allow for DistillableVit to still run predictions

2021-01-11 10:49:14 -08:00

efficient.py

allow for mean pool with efficient version too

2020-12-23 18:15:40 -08:00

vit_pytorch.py

offer a way to use mean pooling of last layer

2020-12-23 17:23:58 -08:00