Training of CLIP in JAX
Project description
CLIP-JAX
This repository is used to CLIP models from 🤗 transformers using JAX.
Installation
pip install -e .
Usage
- Use dataset/prepare_dataset.ipynb to prepare your dataset.
- Train the model with training/train_clip.py.
Supported downstream tasks
- Image classification with
FlaxCLIPVisionModelForImageClassification
TODO
- Add guides
- Add pre-trained models
- Add more downstream tasks
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
clip_jax-0.0.1.tar.gz
(22.7 kB
view hashes)
Built Distribution
Close
Hashes for clip_jax-0.0.1-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | dc464b7693d2ebabd51324bb40a3455c571afaad98a7d4b9c6817273d60afcc9 |
|
MD5 | 7529d75466c864338eb7a7ba90590ea9 |
|
BLAKE2b-256 | 1daa66e7527e9b5920bade774177d59cb3415b01747d26d2832a670d08734f3e |