Pokemon Classifier

🚀 寶可夢分類器

本模型是基於 google/vit-base-patch16-224-in21k 在寶可夢分類數據集上進行微調的版本。它在評估集上取得了以下結果：

• 損失值：5.3367
• 準確率：0.0109

📚 詳細文檔

模型信息

評估結果

該模型名為 pokemon-classifier，在圖像分類任務中，於 pokemon-classification 數據集的測試集上進行評估，準確率為 0.01092896174863388。

訓練過程

訓練超參數

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

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

訓練結果

框架版本

• Transformers 4.35.2
• Pytorch 2.1.0+cu121
• Datasets 2.15.0
• Tokenizers 0.15.0

📄 許可證

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