Platzi Vit Model Elyager

🚀 platzi-vit-model-elyager

該模型是基於 google/vit-base-patch16-224-in21k 在 beans 數據集上微調得到的圖像分類模型，在評估集上取得了較高的準確率。

🚀 快速開始

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

• 損失值：0.0677
• 準確率：0.9774

📚 詳細文檔

模型信息

訓練過程

訓練超參數

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

• 學習率（learning_rate）：0.0002
• 訓練批次大小（train_batch_size）：8
• 評估批次大小（eval_batch_size）：8
• 隨機種子（seed）：42
• 優化器（optimizer）：Adam，betas=(0.9, 0.999)，epsilon=1e-08
• 學習率調度器類型（lr_scheduler_type）：線性
• 訓練輪數（num_epochs）：4

訓練結果

框架版本

• Transformers 4.25.1
• Pytorch 1.13.0+cu116
• Datasets 2.8.0
• Tokenizers 0.13.2

📄 許可證

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