Vit Base Patch16 224 Chest X Ray

🚀 vit-base-patch16-224-chest-x-ray

本模型是 google/vit-base-patch16-224 在胸部X光分類數據集上的微調版本，可用於胸部X光圖像的分類任務。

🚀 快速開始

此模型是基於胸部X光分類數據集對 google/vit-base-patch16-224 進行微調得到的。

🔧 技術細節

訓練超參數

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

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

框架版本

• Transformers：4.27.4
• Pytorch：2.0.0+cu118
• Datasets：2.11.0
• Tokenizers：0.13.3

📄 許可證

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