Resnet18 Catdog Classifier

🚀 殘差網絡18（ResNet18）貓狗分類器

本項目基於預訓練的殘差網絡18（ResNet18）模型，在特定的貓狗圖像數據集上進行微調，構建了一個貓狗圖像分類模型，可準確識別圖像中的貓和狗。

🚀 快速開始

本模型是基於預訓練模型 microsoft/resnet-18，在 自定義 數據集上微調得到的。該模型使用了從Kaggle獲取的 “貓狗分類” 數據集進行構建。在模型構建過程中，使用了PyTorch框架和預訓練的ResNet-18模型，採用微調的方法在數據集上進行訓練。

📚 詳細文檔

訓練結果

訓練過程

訓練超參數

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

• 損失函數：交叉熵損失（CrossEntropyLoss）
• 優化器：AdamW
• 學習率：0.0001
• 批次大小：16
• 訓練輪數：10

框架版本

• Transformers 4.33.2
• Pytorch 2.0.1+cu118
• Datasets 2.14.5
• Tokenizers 0.13.3

📄 許可證

本項目採用 Apache 2.0 許可證。