affjljoo3581/deit3-jax

Jax/Flax implementation of DeiT and DeiT-III (ViT)

语言：Shell · GitHub 星标：18 · Fork：6

话题：deep-learning · flax · image-classification · jax · machine-learning · timm · tpu · transformer

在 GitHub 上查看原仓库