Open VarunS1997 opened 9 months ago
One additional overhead work is needed.
please add keras_cv/models/feature_extractor/coca to this file https://github.com/keras-team/keras-cv/blob/master/.kokoro/github/ubuntu/gpu/build.sh to line 72 and 86
PS: we will fix this overhead soon, but in the mean time this is what we need to do to make sure the large GPU tests run.
@VarunS1997 will you be completing this one?
What does this PR do?
Implements the work done in the "CoCa": Contrastive Captioners are Image-Text Foundation Models" (https://arxiv.org/pdf/2205.01917.pdf).
This PR requires:
Before submitting