Violation Classification Bantai Vit Withes

🚀 違規分類 - 板臺視覺變換器（帶早停機制）

本模型是基於google/vit-base-patch16-224-in21k在image_folder數據集上微調得到的。它在評估集上取得了以下成績，展現出了良好的性能和效率：

• 評估損失：0.2234
• 評估準確率：0.9592
• 評估運行時間：64.9173
• 每秒評估樣本數：85.37
• 每秒評估步數：2.68
• 訓練輪數：227.72
• 訓練步數：23000

🚀 快速開始

本模型是在image_folder數據集上對google/vit-base-patch16-224-in21k進行微調的版本。它在評估集上取得了一系列優秀的結果，如低評估損失、高評估準確率等。

🔧 技術細節

訓練超參數

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

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

框架版本

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

📄 許可證

本模型採用Apache - 2.0許可證。