Added feature return to vit models.

martinsbruveris / tensorflow-image-models

TensorFlow port of PyTorch Image Models (timm) - image models with pretrained weights.

https://tfimm.readthedocs.io/en/latest/

Apache License 2.0

286 stars 25 forks source link

Closed martinsbruveris closed 2 years ago

martinsbruveris commented 2 years ago

Adding functionality to return intermediate features from models