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 許可證。