Skip to content

Instantly share code, notes, and snippets.

@radi-cho

radi-cho/gmlp_init.py

Created September 10, 2022 20:03
Show Gist options
  • Save radi-cho/79595e9a02cbcf689cb73ab4581841c9 to your computer and use it in GitHub Desktop.
Save radi-cho/79595e9a02cbcf689cb73ab4581841c9 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gmlp_init.py
import tensorflow as tf
from tensorflow import keras
from tensorflow.keras import layers
class gMLPLayer(layers.Layer):
def __init__(self, num_patches, embedding_dim, dropout_rate, *args, **kwargs):
super(gMLPLayer, self).__init__(*args, **kwargs)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment