Ak Vit Base Patch16 224 In21k Image Classification

🚀 ak-vit-base-patch16-224-in21k-image_classification

本模型是在image_folder數據集上對 google/vit-base-patch16-224-in21k 進行微調後的版本。它在評估集上取得了出色的效果，能夠精準地完成圖像分類任務。

🚀 快速開始

此模型是 google/vit-base-patch16-224-in21k 在image_folder數據集上的微調版本。它在評估集上取得了以下結果：

• 損失值：3.1599
• 準確率：1.0

🔧 技術細節

訓練超參數

訓練過程中使用了以下超參數：

• 學習率：5e-05
• 訓練批次大小：32
• 評估批次大小：32
• 隨機種子：42
• 梯度累積步數：4
• 總訓練批次大小：128
• 優化器：Adam（β1 = 0.9，β2 = 0.999，ε = 1e-08）
• 學習率調度器類型：線性
• 學習率調度器熱身比例：0.1
• 訓練輪數：3

訓練結果

框架版本

• Transformers 4.18.0
• Pytorch 1.10.0+cu111
• Datasets 2.1.0
• Tokenizers 0.12.1

📄 許可證

本項目採用 Apache-2.0 許可證。