Cat Vs Dog Classification

🚀 貓狗分類模型

本模型用於解決貓狗圖像分類問題，基於預訓練模型在貓狗數據集上微調，具有高準確率，能有效區分貓和狗的圖像。

🚀 快速開始

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

• 評估損失（eval_loss）：0.0226
• 評估準確率（eval_accuracy）：0.9944
• 評估運行時間（eval_runtime）：38.0768
• 每秒評估樣本數（eval_samples_per_second）：61.481
• 每秒評估步數（eval_steps_per_second）：1.943
• 訓練輪數（epoch）：1.2
• 訓練步數（step）：705

📚 詳細文檔

訓練超參數

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

框架版本

• Transformers：4.37.2
• Pytorch：2.1.0+cu121
• Datasets：2.17.1
• Tokenizers：0.15.2

📄 許可證

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