Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add clipUtils #2473

Merged
merged 17 commits into from
Aug 13, 2024
Prev Previous commit
Next Next commit
import layer fix
  • Loading branch information
sachinprasadhs committed Aug 12, 2024
commit 7440b656e1f46863425a879273d0c47bafa183be
4 changes: 2 additions & 2 deletions keras_cv/src/layers/vit_layers.py
Original file line number Diff line number Diff line change
@@ -15,14 +15,14 @@
import math

import tensorflow as tf
from keras_cv.src.backend.keras import layers
from keras_cv.src.backend import keras
from keras_cv.src.backend import ops

from keras_cv.src.api_export import keras_cv_export


@keras_cv_export("keras_cv.layers.PatchingAndEmbedding")
class PatchingAndEmbedding(layers.Layer):
class PatchingAndEmbedding(keras.layers.Layer):
"""
Layer to patchify images, prepend a class token, positionally embed and
create a projection of patches for Vision Transformers
Loading