Cat Dog Classifier With Small Datasest

🚀 小型數據集貓狗分類器

本模型是基於圖像文件夾數據集對 microsoft/resnet-50 進行微調後的版本。它在評估集上取得了以下成績：

• 損失值：0.1369
• 準確率：0.95

📦 模型信息

📚 模型索引

• 名稱：cat_dog_classifier_with_small_datasest
  • 結果：
    • 任務：
      • 名稱：圖像分類
      • 類型：圖像分類
    • 數據集：
      • 名稱：imagefolder
      • 類型：imagefolder
      • 配置：默認
      • 劃分：訓練集
      • 參數：默認
    • 評估指標：
      • 名稱：準確率
      • 類型：準確率
      • 值：0.95

🔧 訓練過程

訓練超參數

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

• 學習率：4e-05
• 訓練批次大小：8
• 評估批次大小：8
• 隨機種子：42
• 優化器：使用 adamw_torch，β值為(0.9, 0.999)，ε值為 1e-08，無額外優化器參數
• 學習率調度器類型：線性
• 學習率調度器熱身步數：500
• 訓練輪數：20

訓練結果

框架版本

• Transformers 4.47.1
• Pytorch 2.4.1+cu121
• Datasets 3.2.0
• Tokenizers 0.21.0