Cat Emotion Classifier

🚀 vit-base-cat-emotions

這是一個基於預訓練模型微調的圖像分類模型，用於識別貓咪的情緒。你可以在線體驗該模型，還能在GitHub倉庫中查看更多細節。

🚀 快速開始

你可以點擊此處在線體驗該模型，並前往GitHub倉庫查看更多詳細信息。

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

• 損失值：1.0160
• 準確率：0.6353

🔧 技術細節

訓練超參數

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

• 學習率：0.0002
• 訓練批次大小：16
• 評估批次大小：8
• 隨機種子：42
• 優化器：Adam（β1 = 0.9，β2 = 0.999，ε = 1e-08）
• 學習率調度器類型：線性
• 訓練輪數：10
• 混合精度訓練：Native AMP

訓練結果

框架版本

• Transformers 4.44.1
• Pytorch 2.2.2+cu118
• Datasets 2.20.0
• Tokenizers 0.19.1

📄 許可證

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