- 博客(1)
- 收藏
- 关注
RSS订阅转载 手把手教你实现PyTorch版ViT：图像分类任务中的Transformer实战
我们首先在我们自定义数据集上进行测试，请保证数据集格式为：接着在训练代码中修改数据集路径并运行下述命令即可开始训练：训练完成后文件内会有一个best_vit.pth以及两个训练的Acc及Loss图用于分析接下来我们运行即可执行推理啦！
2025-06-18 03:35:28
35
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人