keras-nlp icon indicating copy to clipboard operation
keras-nlp copied to clipboard

Add ViT

Open divyashreepathihalli opened this issue 1 year ago • 1 comments

This task is part of the KerasHub project ( Consolidating all of KerasCV and KerasNLP models in to one place) Models from KerasCV will be added to KerasNLP in KerasNLP infrstructure style. The development is now under the keras-hub branch in KerasNLP - https://github.com/keras-team/keras-nlp/tree/keras-hub

Steps to complete this Task

  • Destination : A model folder here https://github.com/keras-team/keras-nlp/tree/keras-hub/keras_nlp/src/models
  • Example PR : https://github.com/keras-team/keras-nlp/pull/1737, please follow the style from here and take a look at the comments on the PR for any design clarifications
  • copy the model over
  • use KerasNLP test routines to update the tests
  • for reformatting the code run shell/api_gen.sh, shell/format.sh and shell/lint.sh
  • The backbone will be added as a model, any downstream task model will be under the same model folder.

Add either @divyashreepathihalli, @mattdangerw or @SamanehSaadat for review of any PR.

divyashreepathihalli avatar Aug 16 '24 22:08 divyashreepathihalli