Detr Resnet 50 Finetuned Cppe5

🚀 detr-resnet-50_finetuned_cppe5

本模型是 facebook/detr-resnet-50 在 cppe - 5 數據集上的微調版本，可在特定場景下提供更精準的預測結果，提升相關任務的性能。

🚀 快速開始

完整的筆記本： https://github.com/MustafaAlahmid/hugging_face_models/blob/main/detr-resnet50-cppe5.ipynb

🔧 技術細節

訓練超參數

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

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

框架版本

• Transformers 4.26.1
• Pytorch 1.13.1
• Datasets 2.9.0
• Tokenizers 0.13.2

📄 許可證

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