mbrukman / clip-jax Public

forked from borisdayma/clip-jax

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Train CLIP models using JAX and 🤗 transformers

Apache-2.0 license

0 stars 7 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 15 Commits
clip_jax		clip_jax
dataset		dataset
training		training
.gitignore		.gitignore
LICENSE.md		LICENSE.md
Makefile		Makefile
README.md		README.md
setup.py		setup.py

Repository files navigation

CLIP-JAX

This repository is used to CLIP models from 🤗 transformers using JAX.

Installation

pip install .

Usage

Use dataset/prepare_dataset.ipynb to prepare your dataset.
Train the model with training/train_clip.py.

About

Train CLIP models using JAX and 🤗 transformers

Apache-2.0 license

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 95.6%
Jupyter Notebook 4.3%
Makefile 0.1%