Skip to content

Instantly share code, notes, and snippets.

@radi-cho

radi-cho/gmlp_train.py

Created September 10, 2022 20:08
Show Gist options
  • Save radi-cho/7b654afdb34950af1bdbb723a6acd7ea to your computer and use it in GitHub Desktop.
Save radi-cho/7b654afdb34950af1bdbb723a6acd7ea to your computer and use it in GitHub Desktop.
Download ZIP
Raw
gmlp_train.py
gmlp_blocks = keras.Sequential(
[gMLPLayer(num_patches, embedding_dim, dropout_rate) for _ in range(num_blocks)]
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment