Vision Transformer with support for patch or hybrid CNN input stage

Collate examples for supervised fine-tuning.

Collate examples for supervised fine-tuning.

Drop paths (Stochastic Depth) per sample (when applied in main path of residual blocks).

CNN Feature Map Embedding Extract feature map from CNN, flatten, project to embedding dim.

Image to Patch Embedding

Dataset for supervised fine-tuning.

Dataset for supervised fine-tuning.

Implement VIT backbone.