Segformer B0 Finetuned Segments Graffiti

🚀 segformer-b0-finetuned-segments-graffiti

該模型是在Adriatogi/graffiti數據集上對nvidia/mit-b0進行微調後的版本。它在評估集上取得了以下結果：

• 損失值（Loss）：0.3250
• 平均交併比（Mean Iou）：0.8048
• 平均準確率（Mean Accuracy）：0.8943
• 總體準確率（Overall Accuracy）：0.8929
• 非塗鴉準確率（Accuracy Not Graf）：0.8830
• 塗鴉準確率（Accuracy Graf）：0.9056
• 非塗鴉交併比（Iou Not Graf）：0.8227
• 塗鴉交併比（Iou Graf）：0.7870

📚 詳細文檔

模型信息

訓練過程

訓練超參數

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

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

訓練結果

框架版本

• Transformers 4.38.2
• Pytorch 2.2.1+cu121
• Datasets 2.18.0
• Tokenizers 0.15.2

📄 許可證

本項目採用其他許可證。