Vit Base Patch16 224 Finetuned Main Gpu 30e Final

🚀 vit-base-patch16-224-finetuned-main-gpu-30e-final

此模型是 google/vit-base-patch16-224 在 imagefolder 数据集上的微调版本。它在评估集上取得了以下结果：

• 损失值：0.0231
• 准确率：0.9940

📚 详细文档

模型信息

训练过程

训练超参数

训练过程中使用了以下超参数：

• 学习率（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，β1=0.9，β2=0.999，ε=1e-08
• 学习率调度器类型（lr_scheduler_type）：线性
• 学习率调度器预热比例（lr_scheduler_warmup_ratio）：0.1
• 训练轮数（num_epochs）：30

训练结果

框架版本

• Transformers：4.26.1
• Pytorch：1.13.1+cu116
• Datasets：2.10.1
• Tokenizers：0.13.2

📄 许可证

本模型采用 Apache-2.0 许可证。