license: apache-2.0 tags:

• 训练生成 datasets:
• 胸部X光分类数据集 metrics:
• 准确率 model-index:
• name: vit-肺炎检测模型 results:
  • task: name: 图像分类 type: 图像分类 dataset: name: 胸部X光分类数据集 type: 胸部X光分类 config: 完整版 split: 验证集 args: 完整版 metrics:
    • name: 准确率 type: 准确率 value: 0.976824034334764

vit-肺炎检测模型

该模型是基于google/vit-base-patch16-224-in21k在胸部X光分类数据集上微调的版本。 在评估集上取得了如下结果：

• 损失值：0.1086
• 准确率：0.9768

模型描述

需要补充更多信息

预期用途与限制

需要补充更多信息

训练与评估数据

需要补充更多信息

训练流程

训练超参数

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

• 学习率：0.0002
• 训练批次大小：64
• 评估批次大小：64
• 随机种子：42
• 优化器：Adam（β1=0.9，β2=0.999，ε=1e-08）
• 学习率调度器类型：余弦退火
• 学习率预热比例：0.25
• 训练轮次：10

训练结果

框架版本

• Transformers 4.26.1
• Pytorch 1.13.1+cu116
• Datasets 2.10.1
• Tokenizers 0.13.2