Violation Classification Bantai Vit Withes

🚀 violation - classification - bantai - vit - withES

このモデルは、[google/vit - base - patch16 - 224 - in21k](https://huggingface.co/google/vit - base - patch16 - 224 - in21k) を image_folder データセットでファインチューニングしたバージョンです。評価セットでは以下の結果を達成しています。

• eval_loss: 0.2234
• eval_accuracy: 0.9592
• eval_runtime: 64.9173
• eval_samples_per_second: 85.37
• eval_steps_per_second: 2.68
• epoch: 227.72
• step: 23000

📚 ドキュメント

トレーニング手順

トレーニングハイパーパラメータ

トレーニング中に以下のハイパーパラメータが使用されました。

• learning_rate: 5e - 05
• train_batch_size: 32
• eval_batch_size: 32
• seed: 42
• gradient_accumulation_steps: 4
• total_train_batch_size: 128
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e - 08
• lr_scheduler_type: linear
• lr_scheduler_warmup_ratio: 0.1
• num_epochs: 500

フレームワークバージョン

• Transformers 4.18.0
• Pytorch 1.10.0+cu111
• Datasets 2.1.0
• Tokenizers 0.12.1

📄 ライセンス

このモデルは Apache - 2.0 ライセンスの下で提供されています。