Code

Implemented Coca architecture#2371

Code

Open

VarunS1997 wants to merge 13 commits intomasterkeras-team/keras-cv:masterfrom model-impl/CoCakeras-team/keras-cv:model-impl/CoCaCopy head branch name to clipboard

+361-3

Commits

Commits on Mar 4, 2024

• Implemented Coca architecture

  VarunS1997

  committed
  20cdf41

• Minor clean-up

  VarunS1997

  committed
  b8c0ba4

• Fixed depth of decoders

  VarunS1997

  committed
  bbe17c4

• Updated config to match args

  VarunS1997

  committed
  202526f

Commits on Mar 5, 2024

• Moved layer definitions to build and added build calls for each layer

  VarunS1997

  committed
  367dd39

• Unabbreviated 'contrastive' and 'captioning'

  VarunS1997

  committed
  80ea7d3

• Improved documentation and added output sizing to call(), also built each layer with expected sizing

  VarunS1997

  committed
  3feacb6

• Lowercased coca model directory and added to kokoro build

  VarunS1997

  committed
  f15408f

Commits on Mar 11, 2024

• Addressed comments by Matt; reformatted as well

  VarunS1997

  committed
  960873f

Commits on Mar 13, 2024

• Addressed comments related to attn pooling size, attn pooling name

  VarunS1997

  committed
  33cff54

Commits on Mar 19, 2024

• Wrote a test for coca saving and loading, which prompted some model changes

  VarunS1997

  committed
  145d7b5

Commits on Mar 26, 2024

• Updated to functional model

  VarunS1997

  committed
  e8623a9

Commits on Mar 28, 2024

• added size inputs for functional model

  VarunS1997

  committed
  c9e1ec1