Vit Base Patch16 224 In21k Finetuned Brain Tumor

🚀 vit-base-patch16-224-in21k-finetuned-brain-tumor

该模型是基于 google/vit-base-patch16-224-in21k 在 imagefolder 数据集上进行微调的版本。它在评估集上取得了出色的效果，能够有效用于图像分类任务，为相关领域提供了高精度的解决方案。

🚀 快速开始

此模型是 google/vit-base-patch16-224-in21k 在 imagefolder 数据集上的微调版本。它在评估集上取得了以下结果：

• 损失值：0.2753
• 准确率：0.9316

📚 详细文档

模型信息

训练过程

训练超参数

训练过程中使用了以下超参数：

• 学习率：5e-05
• 训练批次大小：40
• 评估批次大小：40
• 随机种子：42
• 梯度累积步数：4
• 总训练批次大小：160
• 优化器：Adam（β1=0.9，β2=0.999，ε=1e-08）
• 学习率调度器类型：线性
• 学习率调度器热身比例：0.1
• 训练轮数：4

训练结果

框架版本

• Transformers：4.26.1
• Pytorch：1.13.1+cu117
• Datasets：2.9.0
• Tokenizers：0.13.2

📄 许可证

本模型采用 Apache-2.0 许可证。