vit-tensorflow

vit-tensorflow copied to clipboard

Cannot fit with the model ViT

3

``` v = ViT( image_size = 256, patch_size = 32, num_classes = 4, dim = 1024, depth = 6, heads = 16, mlp_dim = 2048, dropout = 0.1, emb_dropout =...

DRedflag

Please provide an example for training the ViT model in new dataset.

ashraful-haque-iasri

Mobile ViT

1

Hi @taki0112 , When running the Mobile ViT python file I receive an error. ```python v = MobileViT( image_size=(256, 256), dims=[96, 120, 144], channels=[16, 32, 48, 48, 64, 64, 80,...

conceptofmind

Is there any way to load the pretrained weights into vision transformer?

hrithickcodes