vision_transformer:基于 JAX/Flax 的视觉Transformer模型微调项目

可用于对预训练的Vision Transformer和MLP-Mixer模型进行微调,适用于图像识别任务。提供多种预训练模型,支持在自定义数据集上训练,代码支持GPU/TPU加速,包含详细使用教程和Colab示例。【此简介由AI生成】

分支14Tags0

项目介绍

可用于对预训练的Vision Transformer和MLP-Mixer模型进行微调,适用于图像识别任务。提供多种预训练模型,支持在自定义数据集上训练,代码支持GPU/TPU加速,包含详细使用教程和Colab示例。【此简介由AI生成】

定制我的领域
11012.63 K1.48 K访问 GitHub